leetcode 1218. 最长定差子序列
问题描述
给你一个整数数组 arr 和一个整数 difference,请你找出 arr 中所有相邻元素之间的差等于给定 difference 的等差子序列,并返回其中最长的等差子序列的长度。
示例 1:
输入:arr = [1,2,3,4], difference = 1
输出:4
解释:最长的等差子序列是 [1,2,3,4]。
示例 2:
输入:arr = [1,3,5,7], difference = 1
输出:1
解释:最长的等差子序列是任意单个元素。
示例 3:
输入:arr = [1,5,7,8,5,3,4,2,1], difference = -2
输出:4
解释:最长的等差子序列是 [7,5,3,1]。
代码
class Solution {
public:
int longestSubsequence(vector<int>& arr, int difference) {
int n = arr.size(),ans = 1;
if(n == 0)return 0;
unordered_map<int,int> table;
vector<int>dp(n,1);
table[arr[0]] = 0;
for(int i = 1; i < n; ++i)
{
int num = arr[i] - difference;
if(table.count(num))//说明之前有可以连成等差数列的值
{
int ind = table[num];//找到该数值的坐标
dp[i] = dp[ind] + 1;
ans = max(dp[i],ans);
}
table[arr[i]] = i;
}
return ans;
}
};
结果:
执行用时:304 ms, 在所有 C++ 提交中击败了76.78%的用户
内存消耗:53.3 MB, 在所有 C++ 提交中击败了47.48%的用户