对局匹配（蓝桥杯）

小明喜欢在一个围棋网站上找别人在线对弈。这个网站上所有注册用户都有一个积分，代表他的围棋水平。

小明发现网站的自动对局系统在匹配对手时，只会将积分差恰好是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

再比如，
样例输入：
10 1
2 1 1 1 1 4 4 3 4 4

样例输出：
8

资源约定：
峰值内存消耗 < 256M
CPU消耗 < 1000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入…” 的多余内容。

所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。

注意: main函数需要返回0
注意: 只使用ANSI C/ANSI C++ 标准，不要调用依赖于编译环境或操作系统的特殊函数。
注意: 所有依赖的函数必须明确地在源文件中 #include ， 不能通过工程设置而省略常用头文件。

提交时，注意选择所期望的编译器类型。

分析

• 把n个元素按照将分数相差为k的用户分成一组，例如第一组就是{0,k,2k,3k…}，第二组就是{1,k+1,2k+1…}，等等。这样每个组的用户不可能和其他组的用户匹配成功，因为它们之间的差不可能为k。
• 在每个分组里面选取尽量多的用户。
  • 用num[i]表示积分为i的用户人数，假设现在第i组有m个不同积分{x,x+k,x+2k,…,x+(m−1)k}，其中x表示该组第一个人的积分，我们用动态规划的方法来选择尽量多的人数。

dp[i] 表示选择前i个积分能获得的最大用户人数，dp方程如下：

• dp[i]=max(dp[i−1],dp[i−2]+num[score])，其中 num[score]表示该组第 i个积分的总人数。

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

int N, K;
int ans;
const int MAX = 100000 + 5;
int val[MAX];
int dp[MAX]; 
int num[MAX]; //不同积分的个数

int main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0);

	cin >> N >> K;
	int x;
	for (int i = 0; i < N; i++){
		cin >> x;
		num[x]++; //数字x的个数
	}
	if (K == 0){ //计算有多少个不同的数字
		for (int i = 0; i <= 100000; i++) {
			if (num[i]) ans++;
		}
	}
	else{
		for (int i = 0; i < K;i++){
			int m = 0; 
			for (int j = i; j <= 100000; j+=K){ 
				val[m++] = num[j]; //该组每个数据的个数
			}
			dp[0] = val[0];
			for (int j = 1; j < m; j++){
				if (j == 1){
					dp[j] = max(dp[0],val[j]);
				}
				else{
					dp[j] = max(dp[j-1],dp[j-2]+val[j]);
				}
			}
			ans += dp[m - 1];
		}
	}
	cout << ans << endl;
	system("pause");
	return 0;
}