【leetcode】最长定差子序列 (map+dp)

给你一个整数数组 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;
    }
};