编程打卡第十天之最长定差子序列

题目描述：

给你一个整数数组 arr 和一个整数 difference，请你找出并返回 arr 中最长等差子序列的长度，该子序列中相邻元素之间的差等于 difference 。

子序列 是指在不改变其余元素顺序的情况下，通过删除一些元素或不删除任何元素而从 arr 派生出来的序列。

示例 1：

输入：arr = [1,2,3,4], difference = 1
输出：4
解释：最长的等差子序列是 [1,2,3,4]。
示例 2：

输入：arr = [1,3,5,7], difference = 1
输出：1
解释：最长的等差子序列是任意单个元素。

提示：

• 1 <= arr.length <= 105
• -104 <= arr[i], difference <= 104

解题思路：等差数列：a[n]=a[n-1]+d;a[n]的长度比a[n-1]多1；所以a[n]的长度可以由a[n-1]得到；由于n可取负数，所以用哈希表存储长度集合；

代码：

#include<iostream>
#include<vector>
#include<unordered_map>
using namespace std;

class Solution {
public:
    int longestSubsequence(vector<int>& arr, int difference) {
        unordered_map <int, int> d;//记录以key结尾的最长等差序列的长度
        int mostD = 0;
        //初始化
        for (int a : arr) {
            d[a] = 0;
        }
        for (int a : arr) {
            d[a] = d[a - difference] + 1;
            mostD = max(d[a], mostD);
        }
        return mostD;
    }
};

int main()
{
    vector<int> arr;
    int n;
    Solution solution;
    while (1) {
        cin >> n;
        arr.push_back(n);
        if (cin.get() == '\n')  break;
    }
    cin >> n;
    for (auto p : arr) {
        cout << p  << endl;
    }
    cout<< solution.longestSubsequence(arr, n);
}