（C++/python）LeetCode 1218 最长定差子序列

题目描述：
给你一个整数数组 arr 和一个整数 difference，请你找出 arr 中所有相邻元素之间的差等于给定 difference 的等差子序列，并返回其中最长的等差子序列的长度。

思路：
定义一个map，记录当前位置下的最长定差子序列
然后正序遍历
最后输出最大的长度

代码如下：
C++

class Solution {
public:
    int longestSubsequence(vector<int>& arr, int difference) {
        if(arr.size()<1)   return 0;
        map<int,int>temp;
        int res=1;
        for(auto& num:arr){
            int pre=num-difference;
            auto iter=temp.find(pre);
            if(iter==temp.end())
            temp[num]=1;
            else{
                temp[num]=temp[pre]+1;
                res=max(res,temp[num]);
            }
        }
        return res;
    }
};

python

class Solution:
    def longestSubsequence(self, arr: List[int], difference: int) -> int:
        d = {}
        for n in arr:
            dif = n - difference
            if dif in d:
                d[n] = d[dif] + 1
            else:
                d[n] = 1
        return max(d.values())