蓝桥杯对局匹配JAVA(动态规划)

问题描述
　　小明喜欢在一个围棋网站上找别人在线对弈。这个网站上所有注册用户都有一个积分，代表他的围棋水平。
　　小明发现网站的自动对局系统在匹配对手时，只会将积分差恰好是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

思路：
由于要求任意两名用户积分分差不等K的最大人数，可将所有人分为K组，求出每组的最大人数再相加，这里先利用Map<积分，人数>存储，再将分组人数存入ArrayList<ArrayList<Integer>>中，再对ArrayList中的每一个ArrayList利用线性规划求得最大人数
分组后动态规划的转台转移方程：dp[i] =Math.max(dp[i-1],dp[i-2]+第i个积分的人数);
相邻的不能在一起，所以状态转移是（前一个积分最大人数，前两个积分最大人数+本积分人数）中的最大值

举例：
比如分组k=3，最大的积分max=10
输入为
8 3
2 10 7 1 4 1 8 5
那么最后的分组应该是
0 0 0 0 (积分为0 3 6 9的人数)
2 1 1 1 (积分为1 4 7 10的人数)
1 1 1 (积分为2 5 8的人数)

public class Main {
	public static void main(String[] args) {
		Scanner s = new Scanner(System.in);
		int N = s.nextInt();
		int k = s.nextInt();
		Map<Integer, Integer> map = new HashMap();
		int max = 0;//max来记录最大积分，方便分组
		for (int i = 0; i < N; i++) {
			int num = s.nextInt();
			if (num > max) {
				max = num;
			}
			Integer count = map.getOrDefault(num, 0);
			map.put(num, count + 1);//map<积分，有此积分人数>
		}
		if (k == 0) {//如果k为0的特殊情况，直接返回map的大小
			System.out.println(map.size());
			return;
		}
		List<ArrayList<Integer>> lists = new ArrayList();
		for (int i = 0; i < k; i++) {
			ArrayList<Integer> list = new ArrayList();
			lists.add(list);
		}
		for (int i = 0; i <= max; i++) {
			ArrayList<Integer> templist = lists.get(i % k);
			templist.add(map.getOrDefault(i, 0));//分组，如果map里没有这个积分的人，存入积分为0
		}
		int res = 0;
		for (int i = 0; i < k; i++) {
			ArrayList<Integer> arrayList = lists.get(i);
			int size = arrayList.size();
			if (size < 2) {
				if (size == 1) {
					res += arrayList.get(0);
				}
			} else {
				int[] dp = new int[size];
				dp[0] = arrayList.get(0);
				dp[1] = arrayList.get(1);
				for (int j = 2; j < size; j++) {
					dp[j] = Math.max(dp[j - 1], dp[j - 2] + arrayList.get(j));
				}
				res += dp[size - 1];
			}
		}
		System.out.println(res);
	}
}