[leetcode] 5025. 最长等差数列

题目

给定一个整数数组 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]。

解题思路

是lc第132场竞赛的问题。

第一眼觉得不难结果比hard做得还久。

status : pos + diff => len

给定一个pos作为结尾 + 步长 可以得到相应的等差数列长度。一个pos可以存在不同的步长，因此对应不同的长度。

考虑到pos是确定的大小为A.size()，所以使用vector，而diff的秩很可能很小，所以用了unordered_map。

 

代码

class Solution {
public:
    int longestArithSeqLength(vector<int>& A) {
        // pos + diff => len
        vector<unordered_map<int, int>> m;
        m.resize(A.size());
        int res = 2;

        for(int i = 1; i < A.size(); i++){
            for(int j = 0; j < i; j++){
                int diff = A[i] - A[j];
                if(m[j].find(diff) == m[j].end()){
                    m[i][diff] = 2;
                }
                else{
                    m[i][diff] = m[j][diff] + 1;
                    res = std::max(m[i][diff], res);
                }
            }
        }
        
        return res;
    }
};