Divisiblity of Differences

原题链接

题意

原题大意为: 求给定的 n 个整数, 是否能从中选取 k 个整数, 使这 k 个整数中任意两个整数之差能够被 m 整除, 如存在则输出 Yes 并输出这 k 个数(答案有多个可任意输出一个); 否则就输出 No .(2 ≤ k ≤ n ≤ 100 000, 1 ≤ m ≤ 100 000)

输入

第一行输入 n, k, m;
第二行输入 n 个整数;

例子

input

3 2 3
1 8 4

output

Yes
1 4 

input

3 3 3
1 8 4

output

No

input

4 3 5
2 7 7 7

output

Yes
2 7 7 

解析

如果两个数之差能被 m 整除, 那么这两个数除以 m 的模相同. 即 $k_i$ % m == $k_2$ % m;
那么问题就简单了, 我们只需要找出 k 个除以 m 有相同的模的数就可以了, 若未找到则输出 No;

以下是我写的代码:

#include <algorithm> 
using namespace std;
const int MAXN = 100000;
int a[MAXN], b[MAXN], n, m, k, c, ma, d, x;
//a[MAXN] 统计模为 i 的数的个数 a[i], b[MAXN]是第i个数b[i], ma确定a数组的边界;
// c 是个数大于等于 k 的相同数的模, d 用来判断是否存在这样的 k 个数;
int main() {
	cin >> n >> k >> m;
	for (int i = 0; i < n; i++) {
		cin >> b[i];
		x = b[i] % m;//求该b[i]除以 m 的模;
		a[x]++;
		ma = max(x, ma);//判断a数组的边界;
	}
	for (int i = 0; i <= ma; i++) {
		if (a[i] >= k) {//判断是否存在 不少于 k 个数除以 m 的模相同;
			cout << "Yes" << endl;
			c = i;
			d = 1;
			break;
		}	
	}	
	if (d) {
		x = 0;
		for (int i = 0; i < n; i++) {
		if (b[i] % m == c) {//找到并输出除以 m 的模等于 c 的数;
			cout << b[i] << ' ';
			x++;
			if (x == k)//输出 k 个数后跳出循环;
				break;
		}	
	}
	}
	else {
		cout << "No" << endl;
	}
	
	return 0;
}