zcmu-1907: k倍区间(前缀和)

1907: k倍区间

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 242  Solved: 66
[Submit][Status][Web Board]

Description

给定一个长度为N的数列，A1, A2, ... AN，如果其中一段连续的子序列Ai, Ai+1, ... Aj(i <= j)之和是K的倍数，我们就称这个区间[i, j]是K倍区间。 

你能求出数列中总共有多少个K倍区间吗？

 

Input

每个数据包含多组输入数据

第一行包含两个整数N和K。(1 <= N, K <= 100000)  
以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)

 

Output

输出一个整数，代表K倍区间的数目。

 

Sample Input

5 2 1 2 3 4 5

Sample Output

6

 

普通做法疯狂TLE，稍微改动一下， 例如有1，2，3，4，5五个数字，

前1个数字前缀和 = 1

前2个 = 1 + 2

前3个 = 1 + 2 +3

...

计算这些前缀和%k的余数，用hash记录下余数，cnt[0]一定能被k整除（因为%k余数是0了嘛），（假设k=7）剩下的比如有余数是2，3，5的，一个余数为2和一个余数为5的数相加后能被7整除（抽屉原理好像是，我再去查下），所以在求个余数的前缀和sum。

 

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn = 1e5+5;
ll sum[maxn], cnt[maxn], n, k;
 
int main(void)
{
    while(cin >> n >> k)
    {
        memset(sum, 0, sizeof(sum));
        memset(cnt, 0, sizeof(cnt));
        ll ans = 0;
        for(int i = 1; i <= n; i++)
        {
            ll t;
            scanf("%lld", &t);
            sum[i] = (sum[i-1]+t)%k;
            ans += cnt[sum[i]];
            cnt[sum[i]]++;
        }
        printf("%lld\n", ans+cnt[0]);
    }
    return 0;
}