最长等差数列_最长定差子序列（LC1218）

起初，我的思路是这样的，首先我把用sort函数将arr排个序，如果difference>0,就从小到大，如果difference<=0，就从大到小排，开数组dp[i]记录以arr[i]结尾的部分有序序列构成的最长等差数列个数，如果arr[i]==arr[j]+difference j∈[0,i-1]，dp[i]=dp[j]+1; s=max(dp[i],s);

但是你很就会发现有问题，那就是我们排完序后的序列已经不再是原序列，当然我们得到的也将不是子序列。以下为最初的代码。例子显然会过不了。

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

bool comp(int a,int b)
    {
        return a>b;
    }
class Solution {
public:
    int longestSubsequence(vector<int>& arr, int difference) {
        int dp[arr.size()]={1},s=1;
        if(difference>0)
            sort(arr.begin(),arr.end());
        else
            sort(arr.begin(),arr.end(),comp);
        for(int i=0;i<arr.size();i++)
            cout<<arr[i]<<" ";
        cout<<endl;
        for(int i=0;i<arr.size();i++)
        {
            for(int j=0;j<i;j++)
            {
                if(arr[i]==(arr[j]+difference))
                {
                    dp[i]=dp[j]+1;
                    s=max(dp[i],s);
                }
            }
        }
        return s;
    }
};

int main()
{
    Solution s=Solution();
    vector<int> arr;
    arr.push_back(1);
    arr.push_back(5);
    arr.push_back(7);
    arr.push_back(8);
    arr.push_back(5);
    arr.push_back(3);
    arr.push_back(4);
    arr.push_back(2);
    arr.push_back(1);
    cout<<s.longestSubsequence(arr,-2)<<endl;
    return 0;
}

后来，考虑动态规划解题，递推关系式如下：

G(num) 表示以元素num为定差子序列结尾的子序列长度

F(k) 前k个元素中最长定差子序列的长度

当前元素num所在子序列长度等于前一个序列数num-difference所在子序列的长度+1

如果num之前没有出现num-difference元素 则G[num-difference]为0

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

class Solution {
public:
    int longestSubsequence(vector<int>& arr, int difference) {
        unordered_map <int, int> dp;
        dp.clear();

        // dp[v]表示以v结尾的同时以difference为公差的最长定差子序列
        int n = arr.size(), res = -1e9;
        for(int i = 0; i < n; i ++) {
            dp[arr[i]] = max(dp[arr[i]], dp[arr[i] - difference] + 1);
            res = max(dp[arr[i]], res);
        }
        for (auto item : dp)
		    cout << item.first << " -> " << item.second << endl;
        return res;
    }
};

int main()
{
    Solution s=Solution();
    vector<int> arr;
    arr.push_back(1);
    arr.push_back(5);
    arr.push_back(7);
    arr.push_back(8);
    arr.push_back(5);
    arr.push_back(3);
    arr.push_back(4);
    arr.push_back(2);
    arr.push_back(1);
    cout<<s.longestSubsequence(arr,-2)<<endl;
    return 0;
}