历届试题 对局匹配

问题描述

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

小明发现网站的自动对局系统在匹配对手时，只会将积分差恰好是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{
	static int n,K,max,min;
	static int[] s;
	static int[] dp;
	public static void main(String[] args){
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		K = sc.nextInt();
		max = 0;
		min = 100010;
		s = new int[100010];
		dp = new int[100010];
		for(int i=0;i<n;i++){
			int a = sc.nextInt();
			if(a>max)max=a; // 因为使用数组标记法，所以得到最大最小值以缩小查询范围，提高运行效率
			if(a<min)min=a;
			s[a]++;
		}
		
		int sum=0;
		if(K==0){ // 当差值为0 时特殊处理  每个积分都只取一个人
			for(int i=min;i<=max;i++)
				if(s[i]!=0)
					sum++;
		} else { // 差值不为0时
			int t;
			for(int i=min;i<=max;i++){
				if(s[i]==0)continue;
				t=1;
				while(s[i+t*K]!=0)t++; // 找到冲突序列并计算该序列长度
				if(t!=1){
					sum+=find(i,i+(t-1)*K);
				}
				else // 该积分段玩家不存在冲突玩家，直接相加即可
					sum+=s[i];
				t = 1;
				while(s[i+t*K]!=0){ // 找过就将对应分段玩家做清零处理
					s[i+t*K] = 0;
					t++;
				}
			}
		}
		System.out.println(sum);
	}
	private static int find(int b, int e) {
		
		dp[b] = s[b]; // 初始化第一个冲突玩家
		dp[b+K] = s[b+K]; // 初始化第二个冲突玩家
		int i = b+2*K;
		for(;i<=e;i+=K){
			//更新从第一到当前最大 积分只要至少隔一个k就行了 所以 本积分人数加上比他小两个k积分的玩家人数
			dp[i] = s[i]+dp[i-2*K];
			dp[i-K] = Max(dp[i-K],dp[i-2*K]);
		}
		return Max(dp[i-K],dp[i-2*K]);
	}
	private static int Max(int a, int b) {
		if(a>b)
			return a;
		else
			return b;
	}
}