Leetcode 1027. 最长等差数列

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]。

思路

• 这道题目可以用动态规划来解决
• 状态设置：当我们遍历到下标为i的元素，我们发现要想写出转移方程， 就需要记录公差， 所以我们的状态设置为dp[i][margin]， 表示以i结尾的且公差为margin的等差数列的最大的长度
• 状态转移：dp[i][margin] = max(dp[i][margin], dp[j][margin] + 1);
• 注意：初始值设置为0， 所以最后res + 1
• 这道题最开始用map竟然timeout， 由于vector下标不能为负数， 所以这道题目的公差范围是[-10000,10000]， 我们可以+10000将公差变化成[0,20000], 用vector就过了

代码

class Solution {
public:
    int longestArithSeqLength(vector<int>& A) {
        int size = A.size();
        vector<vector<int>> dp(size, vector<int>(20001, 0));
        int res = 0;

        for (int i = 0;i < size;++i) {
            for (int j = 0;j < i;++j) {
                int margin = A[i] - A[j] + 10000;
                dp[i][margin] = max(dp[i][margin], dp[j][margin] + 1);
                res = max(res, dp[i][margin]);
            }
        }

        return res + 1;
    }
};