试题 历届试题 对局匹配（动态规划）保证简单

问题描述
　　小明喜欢在一个围棋网站上找别人在线对弈。这个网站上所有注册用户都有一个积分，代表他的围棋水平。
　　小明发现网站的自动对局系统在匹配对手时，只会将积分差恰好是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,就比如：a[i+k(n-1)], a[i+kn], a[i+k(n+1)] 时候，模拟是做不了的，同时这个给了我思路，我可以通过分组的动态规划来写，及分组公式为 i+kn.(0<=i<k, i+kn<maxx)
当k=2，maxx=10 的分组情况：
i=0: 0 2 4 6 8 10
i=1: 1 3 5 7 9
状态转移方程为dp[j]=max(dp[j-1],dp[j-2]+cnt[j]), dp[j-1]=a[i+k(n-1)], dp[j-2]=a[i+k(n-2)], cnt[j]=a[i+kn].

#include <iostream>
#include<math.h>
#include <stdio.h>
using namespace std;
#define max(a,b) ((a) > (b) ? (a) : (b))
#define min(a,b) ((a) < (b) ? (a) : (b))
const int maxn = 100000+10;//范围要包含数据范围
int a[maxn];
int dp[maxn];
int cnt[maxn];
int n,k;
int main(){
	cin >> n >> k;
	int maxx = -1;
	int ans = 0;
	for (int i = 0; i < n; i++) {
		int b;
		cin >> b;
		maxx = max(maxx, b);//找到最大的数，这样遍历就不用每次遍历到maxn;
		a[b]++;
		//cout << a[b] << endl;
	}
	if (k == 0) {//当k=0，不用分组，只要不为0 就加一
		for (int i = 0; i <= maxx; i++) {
			if(a[i]!=0)
				ans ++;
		}
	}
	else {
		/*
		   分组记分为 i+kn 其中i:0 到 k-1，且i+kn<=最大的数
		*/
		for (int i = 0; i < k; i++) {//控制i
			int y = 0;
			for (int j = i; j <= maxx; j+=k) {//控制n
				//cout << j << endl;
				cnt[y++] = a[j];//把分组后的每个数值存储在cnt里面 cnt[0]=a[i+0*k],cnt[1]=a[i+1*k]以此类推
			}
			dp[0] = cnt[0];//把dp[0]初始化cnt[0]
			for (int j = 1; j < y; j++) {
				if (j == 1) {
					dp[j] = max(dp[0], cnt[j]);
					//cout << dp[j] << endl;
				}
				else {//相邻的两组不能放在一起，dp[j]=max( dp[i-1],dp[i-2]+cnt[j])
					dp[j] = max(dp[j - 1], dp[j - 2] + cnt[j]);
				}
			}
			ans += dp[y - 1];
		}
	}
	
	//cout << endl;
	cout << ans << endl;
	return 0;
}