开始的思路是:可以将每个数字之间间隔是K的拿取出来,然后形成一个等差数列,然后再每一个这样的等差数列上进行动态规划,比如我们得到的某一个k为2的等差数列是{2,4,6,8},然后每一个数字对应的数量是{2,1,1,2},然后就转换成了一个背包问题,选或者不选当前的数,不选的话,那么dp[i] = dp[i-1],选的话,那么dp[i-1]就不能选了,也就是dp[i] = dp[i-2] + num[i],最终我们的状态转移方程就是dp[i] = max(dp[i-1],dp[i-2] + num[i]),但是我还是优化不出来,总是有一组数据是时间超限的.....
我们先看一下时间超限的程序:
//不怕别人比你聪明,就怕别人比你聪明还比你努力
#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include <set>
#include <stack>
#include <map>
#include<vector>
#include <iterator>
#define INF 0x3f3f3f3f
using namespace std;
const int MAXN = 101000;
vector<int> edge[MAXN];
int n,k,m;
set<int> s;
int num[MAXN];
int value[MAXN];
int dp[MAXN];
int temp[MAXN];
bool vis[MAXN];
int main()
{
scanf("%d%d",&n,&k);
memset(dp,0,sizeof(dp));
memset(vis,false,sizeof(vis));
memset(value,0,sizeof(value));
for(int i =1 ; i <= n; i++)
{
scanf("%d",&m);
s.insert(m); //set用来去重
value[m] ++; //记录每一个数的数量
}
int m = 0;
set<int>::iterator it = s.begin();
for(;it!=s.end();it++) //将去完重的数放到数组中
num[m++] = *it;
sort(num,num+m);
int max_num = num[m-1];
if(k == 0) //当k==0是答案就是有多少不重复的数字
{
printf("%d\n",s.size());
}
else
{
int ans = 0;
for(int i = 0; i < m; i ++)
{
if(!vis[i])//然后向后寻找差是k的倍数的相邻的数值
{
int x = num[i],index = 0;
temp[index++] = x;
vis[i] = true;
for(int j = i+1; j < m; j ++)
{
if(num[j] - x == k)//把持每一个距离都是k
{
x = num[j];//更新上一个数字
temp[index++] = x;
vis[j] = true;
}
}
for(int j = 0; j < index; j ++)
dp[j+1] = value[temp[j]]; //将数据存放在1-index中,空着1的位置,为了后面j-2方便
for(int j = 1; j <= index; j ++) //状态转移过程
dp[j] = max(dp[j-1],dp[j-2] + dp[j]);
ans += dp[index];
}
}
printf("%d\n",ans);
}
}
后来看了以为大神的博客:https://www.cnblogs.com/flyawayl/p/8305203.html
思路基本上是相同的,但是他将这个数据分为了k个数组,而且每个里面都是间距为k的等差数列,并且还不用特意地去挑取特定的数字......