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;
}
};