给你一个整数数组 arr 和一个整数 difference,请你找出并返回 arr 中最长等差子序列的长度,该子序列中相邻元素之间的差等于 difference 。
子序列 是指在不改变其余元素顺序的情况下,通过删除一些元素或不删除任何元素而从 arr 派生出来的序列。
示例 1:
输入:arr = [1,2,3,4], difference = 1
输出:4
解释:最长的等差子序列是 [1,2,3,4]。
方法一:dp+哈希表
通过观察我们不难发现 最长定差子序列的每一项的长度与其前一项的长度息息相关 。故我们可以用 HashMap 储存不同位置上的查找的值,以 虑前 i 个数(第 i 个数必选)时,得到的最长定差子序列长度 为状态定义得出动态规划方程。
public int longestSubsequence(int[] arr, int difference) {
HashMap<Integer, Integer> map = new HashMap<>(1 << 3);
for (int value : arr) map.put(value, map.getOrDefault(value-difference, 0) + 1);
return map.values().stream().max((o1, o2) -> o1 >= o2 ? o1 > o2 ? 1 : 0 : -1).get();
}