动态规划—最长等差数列（leetcode 1027）

题目描述

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

回想一下，A 的子序列是列表 其中 。并且如果 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

问题分析

定义状态：dp[i][j]为以第i位数为结尾且公差为j的最长等差子序列长度。

从前向后遍历数组：

    对于当前的数A[i]，枚举数组中的前i位的元素，记为A[j]；
    计算这两个数的公差，即

    diff = A[i] - A[j]；
    尝试将A[i]接到以A[j]结尾的公差为diff的最长等差数列中，即

    dp[i][diff] = max(dp[i][diff], dp[j][diff] + 1);
    若没有以A[j]结尾的公差为diff的最长等差数列，显然(A[i],A[j])即为公差为diff的最长等差数列，因此

    dp[i][diff] = max(dp[i][diff], dp[j][diff] == 0 ? 2 : dp[j][diff] + 1)。

代码

class Solution {
public:
    int longestArithSeqLength(vector<int>& nums) {

        int n = nums.size();
        int ans = 0;
        int maxValue = *max_element(nums.begin(), nums.end());

        vector<vector<int>> dp(n, vector<int>(maxValue*2+2,0)); // maxValue*2+2 即diff向右偏移maxValue*2+2以确保其不为负数
        
        //  dp[i][diff]：以第i位数为结尾且公差为diff的最长等差子序列长度

        for(int i = 0; i <n; ++i) {
            for(int j = 0; j < i; ++j) {
                int diff = nums[i] - nums[j] + maxValue;
                dp[i][diff] = max(dp[i][diff], dp[j][diff] == 0 ? 2 : dp[j][diff] + 1);
                ans = max(ans, dp[i][diff]);
            }
        }
        return ans;
    }
};

复杂度分析

时间复杂度：O(n^2)

空间复杂度：O(n*max(nums))