2021-11-05 LeetCode每日一题 1218. 最长定差子序列

问题描述：

给你一个整数数组 arr 和一个整数 difference，请你找出并返回 arr 中最长等差子序列的长度，该子序列中相邻元素之间的差等于 difference 。

子序列 是指在不改变其余元素顺序的情况下，通过删除一些元素或不删除任何元素而从 arr 派生出来的序列。

示例 1：

输入：arr = [1,2,3,4], difference = 1
输出：4
解释：最长的等差子序列是 [1,2,3,4]。

示例 2：

输入：arr = [1,3,5,7], difference = 1
输出：1
解释：最长的等差子序列是任意单个元素。

示例 3：

输入：arr = [1,5,7,8,5,3,4,2,1], difference = -2
输出：4
解释：最长的等差子序列是 [7,5,3,1]。

提示：

1 <= arr.length <= 10^5
-104 <= arr[i], difference <= 10^4

解决方案：

利用字典，从第一个元素开始遍历，直到结束，找到最大的子序列长度，时间复杂度：O(n)。

class Solution:
    def longestSubsequence(self, arr: List[int], difference: int) -> int:
        l = len(arr)
        max_len = 1
        dic = {}
        for i in range(l):
            if arr[i] - difference in dic:
                dic[arr[i]] = dic[arr[i]-difference] + 1
                max_len = max(max_len, dic[arr[i]])
            else:
                dic[arr[i]] = 1
        return max_len