题目
给你一个整数数组 arr 和一个整数 difference,请你找出 arr 中所有相邻元素之间的差等于给定 difference 的等差子序列,并返回其中最长的等差子序列的长度。
示例
输入:arr = [1,5,7,8,5,3,4,2,1], difference = -2 输出:4 解释:最长的等差子序列是 [7,5,3,1]。
思路及代码
方法一:
动态规划,用dp[i]记录以i结尾的等差数组最大长度,代码很简短。
public int longestSubsequence(int[] arr, int difference) {
HashMap<Integer, Integer> map = new HashMap<>();
int res = -1;
for (int value : arr){
int temp = map.getOrDefault(value - difference, 0) + 1;
map.put(value, temp);
res = Math.max(temp, res);
}
return res;
}
方法二(超时):
动态规划,dp[i]记录以第i个数字结尾的等差数组最大长度。
遍历数组arr,当遍历第i个数字的时候,使用指针p向前遍历,如果arr[i]-arr[p]==difference,则dp[i] = dp[p]+1。
public int longestSubsequence(int[] arr, int difference) {
int n = arr.length;
int[] dp = new int[n];
int res = 1;
for (int i = 0; i <n; i++){
dp[i] =1;
int p = i-1;
while (p >= 0 && arr[i] - arr[p] != difference) p--;
if (p >= 0){
dp[i] = dp[p] + 1;
res = Math.max(res, dp[i]);
}
}
return res;
}