力扣 1218. 最长定差子序列 哈希 dp

https://leetcode-cn.com/problems/longest-arithmetic-subsequence-of-given-difference/

思路：其实和最长上升子序列是很像的，考虑$d{p}_i$表示以$a_i$结尾的最长等差子序列的长度，那么当$j<i$且$a_j=a_i+d$时有$d{p}_i=max(d{p}_i,d{p}_j+1)$，但是这样做的复杂度是$O(n^2)$的，喜提TLE。我们可以仔细观察一下转移方程，其实时间主要浪费在寻找$a_j=a_i+d$上，用哈希优化一下这个过程不就行了么。可以用$d{p}_i$表示以$i$结尾的最长等差子序列的长度，那么对于任意一个元素$e$，有$d{p}_e=d{p}_{e-d}+1$。

class Solution {
public:
    int longestSubsequence(vector<int>& arr, int difference) {
        unordered_map<int,int> dp;
        int ans=0;
        for(int ele:arr)
        {
            dp[ele]=dp[ele-difference]+1;
            ans=max(dp[ele], ans);
        }
        return ans;
    }
};