蓝桥杯——对局匹配

题目

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

小明发现网站的自动对局系统在匹配对手时，只会将积分差恰好是 K 的两名用户匹配在一起。如果两人分差小于或大于 K，系统都不会将他们匹配。

现在小明知道这个网站总共有 N 名用户，以及他们的积分分别是 A 1,A 2,⋯AN 。

小明想了解最多可能有多少名用户同时在线寻找对手，但是系统却一场对局都匹配不起来(任意两名用户积分差不等于 K)？

输入描述

第一行包含两个整数 N,K。

第二行包含 N 个整数A1,A2,⋯AN。
其中，1≤N≤10^5, 0≤Ai≤10 ^5, 0≤K≤10 ^5。

输出描述
输出一个整数，代表答案。

输入
10 0
1 4 2 8 5 7 1 4 2 8

输出
6

思路

创建一个数组cnt，cnt[i] = 2 表示积分为i的人数为2 ，k为0则选择不相同的数即可，k不为0则分组，循环创建一个数组val（val数组里面存放的是cnt），每个相差为k，则选择这个数组里面的一个最大值（表示这些用户同时在线寻找对手的最大人数），进入下一个循环前先加在和上（下一次循环val数组里面的数不同），可以计算用户同时在线寻找对手的最大人数

#include<iostream>
#include <vector>
//或者以上这两句换成#include <bits/stdc++.h>  
using namespace std;
const int N = 1e5+5;
int main()
{
	int n,k;
	int mx = 0; //最大的积分 
	cin>>n>>k;
	vector<int>a(n);
	vector<int>cnt(N);//cnt[i] = 2 表示积分为i的人数为2 
	for(int i = 0;i < n;i++)
	{
		cin>>a[i];
		cnt[a[i]]++;
		mx = max(mx,a[i]);
	}
	if(!k)//如果k为0，选择不相同的数即可
	{
		int ans = 0;
		for(int i = 0;i <= N;i++)
		{
			if(cnt[i] != 0)
			  ans++;
		}
		cout<<ans;
		return 0;
	}
	int ans = 0; // k不为0时 
	for(int i =0;i < k;i++)
	{
		vector<int>val(N);//和cnt数组相似，但是每个相差为k 
		vector<int>dp(N);
		//val中最大的一个数，因为val相差为k，只能其中选一个 
		int num = 0;
		for(int j = i;j <= mx;j += k)
		{
			val[num] = cnt[j];
			num++;
		}
		dp[0] = val[0];
		for(int j = 1;j < num;j++)
		{
			if(j == 1) dp[1] = max(val[0],val[1]);
			else
			{
				dp[j] = max(dp[j-1],dp[j-2]+val[j]);
			}
		}
		ans += dp[num-1];
	}
	cout<<ans;
	return 0;
}

• 总结：关键是循环创建val数组分组，每个数相差为k，则求的是相差为k那就是val数组里面的一个最大数，循环遍历求和