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 是等差的。

思路:
dp[i][j]表示A[i]的前面公差爲j的數量
則若A[X]-A[i] == j 則dp[x][j] = dp[i][j]+1
再維護一個最大序列
code:

class Solution:
    def longestArithSeqLength(self, A: List[int]) -> int:
        max_l = 0
        diff_dict = {}
        for i in range(0, len(A)):
            diff_dict[i] = {}
            for j in range(0, i):
                diff = A[i] - A[j]
                if diff not in diff_dict[j]:
                    diff_dict[i][diff] = 2
                else:
                    diff_dict[i][diff] = diff_dict[j][diff] + 1
                if diff_dict[i][diff] > max_l:
                    max_l = diff_dict[i][diff]
        return max_l
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章