Leetcode 1027. 最長等差數列

題目描述

給定一個整數數組 A，返回 A 中最長等差子序列的長度。

回想一下，A 的子序列是列表 A[i_1], A[i_2], ..., A[i_k] 其中 0 <= i_1 < i_2 < ... < i_k <= A.length - 1。並且如果 B[i+1] - B[i]( 0 <= i < B.length - 1) 的值都相同，那麼序列 B 是等差的。

示例 1:

輸入：[3,6,9,12]
輸出：4
解釋：
整個數組是公差爲 3 的等差數列。

示例 2:

輸入：[9,4,7,2,10]
輸出：3
解釋：
最長的等差子序列是 [4,7,10]。

示例 3:

輸入：[20,1,15,3,10,5,8]
輸出：4
解釋：
最長的等差子序列是 [20,15,10,5]。

解法

以每個元素作爲等差數列終點，計算出該數列長度，選擇出最大值返回即可。

不妨在每個位置上，以字典結構保存該位置元素與前面每個位置上元素的差值，與對應該差值的數列長度。

class Solution:
    def longestArithSeqLength(self, A: List[int]) -> int:
        ret,arr=1,[{} for i in range(len(A))]
        for i in range(1,len(A)):
            for j in range(i):
                d=A[i]-A[j]
                n=arr[j].get(d,1)+1
                arr[i][d]=n
            ret=max(ret,max(arr[i].values()))
        return ret