试题 历届试题 对局匹配
资源限制
时间限制:1.0s 内存限制:256.0MB
问题描述
小明喜欢在一个围棋网站上找别人在线对弈。这个网站上所有注册用户都有一个积分,代表他的围棋水平。
小明发现网站的自动对局系统在匹配对手时,只会将积分差恰好是K的两名用户匹配在一起。如果两人分差小于或大于K,系统都不会将他们匹配。
现在小明知道这个网站总共有N名用户,以及他们的积分分别是A1, A2, … AN。
小明想了解最多可能有多少名用户同时在线寻找对手,但是系统却一场对局都匹配不起来(任意两名用户积分差不等于K)?
输入格式
第一行包含两个个整数N和K。
第二行包含N个整数A1, A2, … AN。
对于30%的数据,1 <= N <= 10
对于100%的数据,1 <= N <= 100000, 0 <= Ai <= 100000, 0 <= K <= 100000
输出格式
一个整数,代表答案。
样例输入
10 0
1 4 2 8 5 7 1 4 2 8
样例输出
6
//dp
#pragma warning(disable:4996)
#include"iostream"
#include"functional"
#include"algorithm"
#include"cstring"
#include"stack"
#include"cmath"
#include"queue"
#include"vector"
#include"map"
typedef long long int ll;
using namespace std;
ll ar[100009],cnt[100009],dp[100009],js[100009];
int main(){
ll m,k,ma=0;
cin>>m>>k;
for(int i=1;i<=m;i++){
cin>>ar[i];
cnt[ar[i]]++;
ma=max(ma,ar[i]);
}
if(k==0){
ll sum=0;
for(int i=0;i<=ma;i++){
if(cnt[i]) sum++;
}
cout<<sum;
}
else{
ll ans=0;
for(int i=0;i<k;i++){
ll o=0;
for(int j=i;j<=ma;j+=k){
js[o++]=cnt[j];
}
dp[0]=js[0];
for(int j=1;j<o;j++){
if(j==1) dp[j]=max(js[j],dp[0]);
else dp[j]=max(dp[j-1],dp[j-2]+js[j]);
}
ans+=dp[o-1];
}
cout<<ans;
}
}