51nod 1103 N的倍数

题目链接

前缀和+抽屉原理

题意：
给了一个长度为$n$的数组，从中任意找到若干个数，它们的和是$n$的倍数。

题解：
这里可以使用抽屉原理，模$n$的余数总共有$n$个。如果出现前缀和模$n$等于0，表明这前$i$个数的和一定是$n$的倍数；反之，如果没有前缀和等于0，那么此时前缀和模$n$的余数只有$n-1$个，因此对于这$n$个数的前缀和来说，一定会有2个前缀和模$n$的余数相同，那么在这之间的所有数的和模$n$一定等于0，也就是$n$的倍数。

实现细节见代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 5e4 + 10;

ll a[MAXN], pre[MAXN];
map<ll, int> last;
int main() 
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		pre[i] = (pre[i - 1] + a[i]) % n;
		if (pre[i] == 0) {
			cout << i << endl;
			for (int j = 1; j <= i; j++) {
				cout << a[j] << endl;
			}
			return 0;
		}
		if (last[pre[i]]) {
			cout << i - last[pre[i]] << endl;
			for (int j = last[pre[i]] + 1; j <= i; j++) {
				cout << a[j] << endl;
			}
			return 0;
		}
		last[pre[i]] = i;
	}
	cout << "No Solution" << endl; // 可以删去这行
    return 0;
}