问题描述:
给你一个整数数组 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