一.问题描述
小明喜欢在一个围棋网站上找别人在线对弈。这个网站上所有注册用户都有一个积分,代表他的围棋水平。
小明发现网站的自动对局系统在匹配对手时,只会将积分差恰好是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
三.输出格式
一个整数,代表答案。
四.问题分析
4.1 如果k为0,我们只需要选择不相同的数即可,即,求n个数中有几个不重复的数。
4.2 如果k不为0,我们采取分组的方式,将n个人分为k组
4.3 对于一个组中,相邻的两个数必不可能选择。
代码实现:
//对局匹配
#include <iostream>
using namespace std;
const int N=1e5+10;
int cnt[N]={0},d[N]={0};
int main(int argc, const char * argv[]) {
ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
int n,k,sum=0,val,MAX=0;
cin>>n>>k;
for(int i=0;i<n;i++){
cin>>val;
cnt[val]++;d[val]++;
if(cnt[val]==1)//统计共有多少不同的积分
sum++;
MAX=max(MAX,val);//记录最大积分
}
if(k==0)//k为0时,结果为所有不同积分的数目
cout<<sum<<'\n';
else{
sum=0;
for(int i=0;i<k;i++){
int j;
for(j=i;j<=MAX;j+=k){
if(j-2*k>=0)
d[j]=max(d[j-k],d[j-2*k]+cnt[j]);
else if(j-k>=0)
d[j]=max(d[j],d[j-k]);
}
sum+=d[j-k];
}
}
cout<<sum<<'\n';
return 0;
}