第八届蓝桥杯C语言B组决赛 对局匹配

标题：对局匹配
小明喜欢在一个围棋网站上找别人在线对弈。这个网站上所有注册用户都有一个积分，代表他的围棋水平。
小明发现网站的自动对局系统在匹配对手时，只会将积分差恰好是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 <xxx>， 不能通过工程设置而省略常用头文件。

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

大致思路是首先统计出每个数的出现次数 然后通过一直加K把出现的数分为total个集合 最后只需要dp计算出各个集合的最大数即可 每个集合之间肯定是互不影响的 总复杂度应该是N的 由于N只有100000 应该不会超时 不知道思路对没

#include <iostream>
#include <vector>
#include <algorithm>
#include <fstream>
using namespace std;
int main() {
	//fstream fin("F://long.txt");
	int num[111111];
	bool used[111111];
	int buf;
	vector<vector<int> >a(100000);
	int total = 0;//集合数量
	int Maxnum = -1;
	int dp[111111];
	memset(num, 0, sizeof(num));
	memset(used, false, sizeof(used));
	int N, K;
	//fin >> N >> K;
	cin >> N >> K;
	for (int i = 0; i < N; i++) {
		//fin >> buf;
		cin >> buf;
		num[buf]++;
	}
	int Max = 0;
	if (K == 0) {//K等于0的情况专门讨论
		for (int i = 0; i <= 100000; i++)
			if (num[i] != 0)
				Max++;
	}
	else {
		for (int i = 100000; i >= 0; i--) {//找出最大数
			if (num[i] != 0) {
				Maxnum = i;
				break;
			}
		}
		for (int i = 0; i <= Maxnum; i++) {//统计出各个连续集合
			if (!used[i] && num[i] != 0) {
				buf = i;
				while (buf <= Maxnum && num[buf] != 0) {
					a[total].push_back(num[buf]);
					used[buf] = true;
					buf += K;
				}
				total++;
			}
		}
		for (int i = 0; i < total; i++) {//dp求出各个集合的最大数然后累加
			dp[0] = a[i][0];
			if (a[i].size() >= 2)
				dp[1] = max(a[i][0], a[i][1]);
			for (int j = 2; j < a[i].size(); j++)
				dp[j] = max(dp[j - 2] + a[i][j], dp[j - 1]);
			Max += dp[a[i].size() - 1];
		}
	}
	cout << Max << endl;
	return 0;
}