蓝桥杯 java 历届试题 对局匹配

历届试题 对局匹配

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　小明喜欢在一个围棋网站上找别人在线对弈。这个网站上所有注册用户都有一个积分，代表他的围棋水平。

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

import java.util.Scanner;
 

 

 
public class Main {
 
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		int []a = new int[100005];
		
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int k = sc.nextInt();
		for (int i = 1; i <= n; i++) 
			a[sc.nextInt()+1]++;
		int ans = 0;				
		//分组后的结果，各个积分的人数
		int []b = new int[n+5];
		//计算最大人数
		int []c = new int[n+5];			
		//如果人数为1
		if(n == 1) {
			ans++;
		}else {
			if(k != 0) {
				//分为k组，每组相邻两个数相差k，
				for (int i = 0; i < k; i++) {
					int m = 0;
					//分组，相差为k的分到一组
					for (int j = i; j <= n; j += k) {
						//分组的赋值
						b[m++] = a[j];	
					}
					c[0] = b[0];
					for (int j = 1; j < m; j++) {
						if(j == 1)c[j] = Math.max(b[j], c[0]);
						else c[j] = Math.max(c[j-2]+b[j], c[j-1]);
					}
					ans += c[m-1];
				}
			}else {
				for (int i = 0; i < n; i++) {
					if(a[i] != 0) {
						ans++;
					}
				}
			}
		}
		System.out.println(ans);
	}
 
}