给你一个整数数组 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]。
提示:
1 <= arr.length <= 10^5
-10^4 <= arr[i], difference <= 10^4
链接:https://leetcode-cn.com/problems/longest-arithmetic-subsequence-of-given-difference
思路分析:
这道题和最长上升子序列类似,但是如果按照最长上升子序列来写两重循环,会TLE.
当我们知道 arr[i] 的时候,我们就知道已经知道上一个状态的值为 arr[i] - difference 了。
因此,我们不需要向最长上升子序列一样一个循环搜所有的下标,我们可以先预处理每个元素所对应的下标。**考虑到会有重复元素出现,我们的下标也需要一个数组来存。**因此用map<int,vector>操作。
class Solution {
public:
int longestSubsequence(vector<int>& arr, int difference) {
int n = arr.size();
int dp[n+2];
memset(dp,0,sizeof(dp));
unordered_map<int,vector<int>> mp;
for(int i = 0;i < n;i++)
{
mp[arr[i]].push_back(i);
}
int ans = 0;
for(int i = 1;i < n;i++)
{
int id = arr[i] - difference;
for(int j = 0;j < mp[id].size();j++)
{
if(mp[id][j] >= i) break;
dp[i] = max(dp[i], dp[mp[id][j]]+1);
}
ans = max(dp[i], ans);
}
return ans+1;
}
};