![40cc52dbade7e34acd28289b44022e33.png](https://i-blog.csdnimg.cn/blog_migrate/798dc386b53429f98cc9d79e529c6fd2.jpeg)
![f4c0e3ae00fb94e627f4975789701092.png](https://i-blog.csdnimg.cn/blog_migrate/ee41491bdf5ad993ff34e7a04922ac15.jpeg)
起初,我的思路是这样的,首先我把用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;
}
![223489c690fc2664017d6c299aa41d1b.png](https://i-blog.csdnimg.cn/blog_migrate/ec546e4446b7bdcbce1d6bad4c31e5da.jpeg)
![5db950b161556eff93c7ba3b50ae186d.png](https://i-blog.csdnimg.cn/blog_migrate/a194a28aa9867721d8c8dfb5a66eb61a.jpeg)
后来,考虑动态规划解题,递推关系式如下:
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;
}
![67847aebf565e0c03bffa578daa355a8.png](https://i-blog.csdnimg.cn/blog_migrate/5510a5eeb8565c520be9bda5bdb62530.jpeg)
![eeba1a2ad56df111457252dc0f61430e.png](https://i-blog.csdnimg.cn/blog_migrate/f6a1e2cacbb31fdf4cadff9fd8dcc612.jpeg)