SCU 4569 【思维 + 前缀和】

传送门
题意: 就是问给定的n个数字中有多个连续子区间, 使得这些子区间的和是给定的一个m的倍数.

思路: 很明显区间和就是想到前缀和, 在%m一下, 那么怎么统计了, 做过HDU5776 的应该知道, 我们可以找到一个相同的那么就有一个区间, 那么这道题这要把所有的区间找出来, 所以我们需要记录一下之前每一个前缀出现的次数, 然后没遇到一个前缀算一次答案即可, 注意这个区间本来就是m的情况, 也就是我们需要事先把0先加一次即可, 处理为当前这个区间和就是m的倍数情况. 因为这道题累加的数比较大, 所以我们需要用map来统计.

AC Code

const int maxn = 3e5+5;
int a[maxn];
ll s[maxn];
map<ll, int>mp;
void solve()
{
    int n, k;
    while(~scanf("%d%d", &n, &k)) {
        for (int i = 1 ; i <= n ; i ++) {
            scanf("%d", a+i);
            s[i] = s[i-1] + a[i];
            s[i] %= k;
        }
        mp.clear();
        ll ans = 0; mp[0]++;
        for (int i = 1 ; i <= n ; i ++) {
            ans += mp[s[i]];
            mp[s[i]]++;
        }
        printf("%lld\n", ans);
    }
}