暴力穷举解法
挨个变量,没什么技术含量
class Solution {
public int longestSubsequence(int[] arr, int difference) {
int max = Integer.MIN_VALUE;
for(int i = 0;i < arr.length;i++){
int item = arr[i];
int length = 1;
for(int j = i + 1;j<arr.length;j++){
if(item+difference==arr[j]){
item = arr[j];
length++;
}
max = Math.max(length,max);
}
}
return max;
}
}
在力扣中无法通过,超出时间限制。
所以我们应该思考新的解决办法,其实类似这种线性表找序列的题,优先应该思考是不是动态规划。
动态规划法
做题之前也应该看一下提示,提示arr[i]的取值在-10000和10000之间,其实就在暗示我们用数组而不是HashMap.
因为数组下标无法表示负数,而负数和正数加上0,一共有40001个数,因此我们给每个数都加20000,以此来达到数组表示全部的目的。
class Solution {
public int longestSubsequence(int[] arr, int difference) {
int[] array = new int[40001];
int max = 0;
for(int i: arr){
array[i + 20000] = array[i + 20000 - difference] + 1;
max = Math.max(array[i+20000],max);
}
return max;
}
}
array[i]表示以i结尾的等差数列的长度,我们从左向右遍历。如果array[i-difference]的值不为0,则说明i左侧有以i-difference为结尾的等差数列,且可以和i组成新的等差数列,因此dp[i]=dp[i-difference]+1,并且在途中统计最长的值。