牛客小白月赛58-C-牛牛

很好的一道模拟题,做起来很舒服

做模拟题重要的还是心态,能静得下心读题

思路:

读完题后,很直观的一个思路就是暴力,但2e5的O(n^2)过不去,怎么优化??

就是转换

sum = ∑a[i]

(sum - S0) % m = 0

结果 v = S0 % m

可以这么转换

sum % m = S0 % m

令sum % m = k

那么就是在序列内找两个数a,b使得

(a % m + b % m) % m = k

暴力就是两层for,怎么优化呢

b % m = k - a % m

注意,这里其实不是k - a % m,因为这样可能小于0,后面要再加x*m

这样就可以O(n)的找了

本题的优化技巧比较常见,就是枚举,因此还是比较简单的,主要还是得静下心读题,读懂了题,实现起来也就得心应手了

// Problem: 牛牛
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/41173/C
// Memory Limit: 524288 MB
// Time Limit: 2000 ms
// Date: 2024-03-21 17:10:21
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
#define endl '\n'
#define int int64_t
#define ld long double
using namespace std;
int a[200005];
void solve() {
	int n, m; cin >> n >> m;
	int sum = 0;
	unordered_map<int, int>p;
	for (int i = 1; i <= n; ++i) {
		cin >> a[i];
		sum += a[i];
		p[a[i] % m]++;
	}
	int k = sum % m;
	for (auto it : p) {
		int l = it.first;
		int val = k - l;
		val = (val + (-val / m + 1) * m) % m;
		if (p.count(val)) {
			if (val == l) {
				if (p[l] >= 2) {
					cout << (k == 0 ? m : k) << endl;
					return;
				}
			}
			else {
				cout << (k == 0 ? m : k) << endl;
				return;
			}
		}
	}
	cout << 0 << endl;
}
signed main() {
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	int t = 1;
	cin >> t;
	while (t--) {
		solve();
	}
	return 0;
}