题目描述:
给你一个整数数组 arr 和一个整数 difference,请你找出并返回 arr 中最长等差子序列的长度,该子序列中相邻元素之间的差等于 difference 。
子序列 是指在不改变其余元素顺序的情况下,通过删除一些元素或不删除任何元素而从 arr 派生出来的序列。
示例:
输入:arr = [1,2,3,4], difference = 1 输出:4 解释:最长的等差子序列是 [1,2,3,4]。
输入:arr = [1,3,5,7], difference = 1 输出:1 解释:最长的等差子序列是任意单个元素。
输入:arr = [1,5,7,8,5,3,4,2,1], difference = -2 输出:4 解释:最长的等差子序列是 [7,5,3,1]。
思路解析:
本题可考虑动态规划算法
针对数组中的任意一个数字i, 使用集合d, 使d[i]纪录满足条件序列的长度。
通过d[i - difference]可找到满足 i - difference 的元素的长度。
代码:(python3)
class Solution:
def longestSubsequence(self, arr: List[int], difference: int) -> int:
d = {}
for a in arr:
d[a] = d.get(a - difference, 0) + 1
print(d[a])
print(d)
return max(d.values())