ABC 367

D. Pedometer

        先断环成链，然后从 n 开始一直到 2n - 1，判断一下前缀和。n^2 做法

 

for (int i = n; i < 2 * n; i ++)
    for (int j = i - n + 1; j < i; j ++)
        if (d[i] - d[j - 1] % m == 0)
            ans ++;

 

        注意到取模的性质，也就是 d [ i ] % m == d [ j - 1 ] % m，因此只需要统计一下模数相等的就可以了。

        注意滑动窗口两端要同时更新。

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 2e5 + 5, INF = 1e18;

int T, n, m, cnt, ans, a[N * 2], d[N * 2];
map<int, int> mp;

signed main()
{
	cin >> n >> m;
	for (int i = 1; i <= n; i ++)
	{
		cin >> a[i];
		a[i + n] = a[i];
	}
	for (int i = 1; i <= 2 * n; i ++)
	{
		d[i] = d[i - 1] + a[i];
		d[i] %= m;
	}
	for (int i = 1; i < n; i ++)
		mp[d[i]] ++;
	for (int i = n; i < 2 * n; i ++)
	{
		if (mp.count(d[i]) != 0)
			ans += mp[d[i]];
		mp[d[i - n + 1]] --;
		mp[d[i]] ++;
	}
	cout << ans;
	return 0;
}