给数组arr和一个差值difference, 不打乱arr中数字的顺序,抽取最长的子序列,使序列中每相邻两个元素的差值为difference.
求满足条件的最长子序列的长度。
思路:
DP
因为差值difference是固定的,每抽取一个元素,它前面的元素也是固定的。
比如现在的元素为arr[ i ], 那么它前面的元素一定是arr[ i ] - difference.
所以到arr[ i ]为止的序列长度也是固定的。
只需要统计到arr[ i ]为止的子序列长度。
记下这些长度的最大值。
dp数组可以用一个hashMap, 保存(arr[ i ], 到arr[ i ]为止的子序列长度),
也可以用数组。
(1)HashMap
public int longestSubsequence(int[] arr, int difference) {
Map<Integer,Integer> dp = new HashMap<>();
int n = arr.length;
int res = 0;
for(int i = 0; i < n; i++) {
dp.put(arr[i], dp.getOrDefault(arr[i]-difference, 0)+1);
res = Math.max(res, dp.get(arr[i]));
}
return res;
}
(2) 数组
直接下标取元素,速度更快,
为了节省存储空间,dp数组长度为arr的最大值-最小值+1.
但是要注意dp中的下标是arr[ i ] - min. 并不是arr的下标 i。
public int longestSubsequence(int[] arr, int difference) {
int min = Integer.MAX_VALUE;
int max = Integer.MIN_VALUE;
int res = 1;
for(int num : arr) {
min = Math.min(min, num);
max = Math.max(max, num);
}
int[] dp = new int[max-min+1];
for(int i = 0; i < arr.length; i++) {
int cur = arr[i] - min;
int pre = cur - difference;
if(pre >= 0 && pre < dp.length) {
dp[cur] = dp[pre] + 1;
if(dp[cur] > res) res = dp[cur];
} else {
dp[cur] = 1;
}
}
return res;
}