【蓝桥杯真题】倍数问题

---

倍数问题（2018年第九届蓝桥杯JavaA组省赛 第九题）

【题目描述】

众所周知，小葱同学擅长计算，尤其擅长计算一个数是否是另外一个数的倍数。但小葱只擅长两个数的情况，当有很多个数之后就会比较苦恼。现在小葱给了你 n 个数，希望你从这 n 个数中找到三个数，使得这三个数的和是 K 的倍数，且这个和最大。数据保证一定有解。

【输入格式】

从标准输入读入数据。
第一行包括 2 个正整数 n, K。
第二行 n 个正整数，代表给定的 n 个数。

【输出格式】

输出到标准输出。
第一行包括 2 个正整数 n, K。
第二行 n 个正整数，代表给定的 n 个数。

【样例输入】

4 3
1 2 3 4

【样例输出】

9

---

测试网址：null

---

个人解法

【解法一】

代码：

import java.util.Arrays;
import java.util.Scanner;

/**
 * 最容易出错的点就是求inp的值，可能会遗漏(p != j && p == i)这个条件
 * 
 * @author HQL
 *
 */
public class Main {

	static int n, k;
	static int[] r;

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		k = sc.nextInt();
		r = new int[n];
		for (int i = 0; i < n; ++i)
			r[i] = sc.nextInt();
		System.out.println(screen());
	}

	/**
	 * 假设k==3时， 找出对3取余为0的、余为1的、余为2的所有数，结果必定在{000， 012， 111， 222}中
	 */
	public static int screen() {
		int[][] li = new int[k][3];
		int[] ind = new int[k];
		int total = 0, res = Integer.MIN_VALUE;
		Arrays.sort(r);

		for (int i = r.length - 1; i >= 0; --i) {
			int p = r[i] % k;
			if (ind[p] < 3) {
				li[p][ind[p]++] = r[i];
				++total;
			}
			if (total == 3 * k)
				break;
		}

		for (int i = 0; i < k; ++i) {
			if (li[i][0] == 0)
				continue;
			for (int j = i; j < k; ++j) {
				int inj = 0, inp = 0, sum = li[i][0];

				if (j == i)
					inj = 1;
				if (li[j][inj] == 0)
					continue;
				sum += li[j][inj];
				
				int p = (k - (sum % k)) % k;
				if (p == i && p == j)
					inp = 2;
				else if (p != j && p == i || p != i && p == j)
					inp = 1;
				if (li[p][inp] == 0)
					continue;
				sum += li[p][inp];
				
				res = Math.max(res, sum);
			}
		}

		return res;
	}

}