leetcode 1027. 最长等差数列

题目

给定一个整数数组 A，返回 A 中最长等差子序列的长度。

回想一下，A 的子序列是列表 A[i_1], A[i_2], …, A[i_k] 其中 0 <= i_1 < i_2 < … < i_k <= A.length - 1。并且如果 B[i+1] - B[i]( 0 <= i < B.length - 1) 的值都相同，那么序列 B 是等差的。

示例 1：

输入：[3,6,9,12]
输出：4
解释：
整个数组是公差为 3 的等差数列。

示例 2：

输入：[9,4,7,2,10]
输出：3
解释：
最长的等差子序列是 [4,7,10]。

示例 3：

输入：[20,1,15,3,10,5,8]
输出：4
解释：
最长的等差子序列是 [20,15,10,5]。

提示：

2 <= A.length <= 2000
0 <= A[i] <= 10000

这道题看起来和最长上升子序列很相似,所以我们也是用相似的思路,整体复杂度为O(n²),看数据的范围也不会超.
首先在每一个维度记录一个和等差项大小相关的数据结构,一般常用vector,这时dp[i][j]=a;表示,从开头到第i个元素差值为j的最长序列长度为a.动态转移方程也很好想了,就是

for(int i=1;i<sz;++i)
        {
   
           for(int j=0;j<i;++j)
           {
   
    }
}

进行遍历,更新查看每一个j的A[i]-A[j]的差值tmp,更新的值是dp[i][tmp],再去看dp[j][tmp]的值为多少,没有这一项的化就天2,表示i,j项,有的化就是加1.
dp[i][tmp]=max(dp[j][tmp]+1,dp[i][tmp]);
dp[i][tmp]=2;
两种情况,由于我们定义的dp[i][tmp]的含义是以i为结尾的最长,所以每一次更新dp都要更新ans值
基本思路就是这样,并且也能过,代码如下:

代码1

class Solution {
   
public:
    int longestArithSeqLength(vector<int>& A) {
   
        int N = A.size();
        vector<vector<int>> dp(N, vector<int>(20010, 1));
        int res = 1;
        for (int i = 0; i < N; i++) {
   
            for (int j = 0; j < i; j++