思想: 本质还是最长子序列的问题,还是借助dp思想
但是这题的不能仅用一个状态,还要用一个状态diff表示差值
dp[i][diff]表示遍历到第i个位置时差为diff的最长等差子序列的长度(也是可以不连续的))
base case:
dp[i][0] =1
状态转移方程:
dp[i][diff] = Max{dp[j][diff]}+1 j=0,1,2…i-1
也就是A[i]可以跟它前面的公差为diff的等差子序列构成一个等差为diff的等差子序列(因为不是求连续的递增子序列)
时间复杂度为O(n),空间复杂度为O(1)
class Solution {
public int longestArithSeqLength(int[] A) {
int len = A.length;
if(len<=0) return 0;
int[][]dp = new int[len][20001];//因为题目给定了0 <= A[i] <= 10000,且公差可以为负,所以留0-10000给公差>=0的用,剩下的10001-20000留给公差为负(diff从-10000....-1)的用
int res = 0;
//遍历状态
for(int i=0;i<len;i++){
//做选择
for(int j=i-1;j>=0;j--){
int diff = A[i]-A[j];
if(diff>=0){
dp[i][diff] = Math.max(dp[i][diff],dp[j][diff]+1);
res = Math.max(res,dp[i][diff]);
}else{
diff+=20001;
dp[i][diff] = Math.max(dp[i][diff],dp[j][diff]+1);
res = Math.max(res,dp[i][diff]);
}
}
}
return res+1;//加1的原因是因为在定义dp数组时没有算上最后递增子序列中最开始的那个元素,手动模拟一下就知道了
}
}