东华oj进阶52 k倍区间

问题描述:

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

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

输入说明 :

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

输出说明 :

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

输入范例 :

5 2
1
2
3
4
5

输出范例 :

6

思路:

这题也是学习的思路，用sum记录下前n项和，mod记下前n项和的不同余数的个数。
如果前 i 项和前 j 项的和相同，那么它们的差就可以构成一段连续序列且除以K的余数为0
因此再计算每项的排列组合数目（C上2下该余数的个数）

踩坑点：

1、没注意题目的数量级，最高数字可达100000*100000，超出了int的上限，需要把sum，res都改成longlong才行，否则会丢结果
2、一开始把余数为0的单独拿出来处理，没有进行排列组合计算，事实上余数为0的也要进行排列组合计算，并且最后要把余数为0的单独加一次

代码实现:

#include<bits/stdc++.h>
using namespace std;

int main()
{
    int N,K;
    long long res=0;
    cin>>N>>K;
    int *arr=new int[N];
    //mod[i]代表前i个元素的最多有多少个k倍区间
    long long *mod=new long long[K];
    //sum[i]代表前i+1个元素和
    long long *sum=new long long[N];

    fill(sum,sum+N,0);
    fill(mod,mod+K,0);

    for(int i=0;i<N;i++)
    {
        cin>>arr[i];
        if(i!=0)
            sum[i]=sum[i-1]+arr[i];
        else
            sum[i]=arr[i];

        mod[sum[i]%K]++;
    }

    for(int i=0;i<K;i++)
    {
        if(mod[i]>0) res+=mod[i]*(mod[i]-1)/2;
    }
    res+=mod[0];

    cout<<res<<endl;
}