P8649 [蓝桥杯 2017 省 B] k 倍区间

[蓝桥杯 2017 省 B] k 倍区间

题目描述

给定一个长度为 $N$ 的数列，$A_1,A_2,\cdots A_N$，如果其中一段连续的子序列 $A_i,A_{i+1},\cdots A_j(i\le j)$ 之和是 $K$ 的倍数，我们就称这个区间 $[i,j]$ 是 $K$ 倍区间。

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

输入格式

第一行包含两个整数 $N$ 和 $K$$(1\le N,K\le 10^5)$。

以下 $N$ 行每行包含一个整数 $A_i$$(1\le A_i\le 10^5)$。

输出格式

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

样例

样例输入

5 2
1  
2  
3  
4  
5

样例输出

6

提示

时限 2 秒, 256M。蓝桥杯 2017 年第八届

Java代码

import java.io.*;
class Main
{
    public static void main(String[] args) throws IOException
    {
        StreamTokenizer in = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
        PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
        in.nextToken();
        int n=(int)in.nval;
        in.nextToken();
        int k=(int)in.nval;
        int[] s = new int[n+1];
        int[] m = new int[k];
        for(int i=1;i<=n;i++)
        {
            in.nextToken();
            s[i]=(int)in.nval%k;     //用一个数组存输入及其前缀和,并对k取模很关键
            s[i]=(s[i-1]+s[i])%k;   //前缀和
        }
        
        long cnt=0;
        m[0]=1;
        for(int i=1;i<=n;i++)
        {
            cnt+=m[s[i]%k];     //从0到i与s[i]同余的
            m[s[i]%k]++;    //把s[i]加进m[]里去
        }
        out.println(cnt);
        out.flush();
    }
}

解析及心得

只用前缀和的时间复杂度为 $O(n^2)$ ，显然不能 AC，这里第23行到第27行将 $i$ 作为要判定区间的右边界， $cnt$ 为 $k$ 倍区间计数器， $m[i]$ 为对 $k$ 取模余数相同的前缀和数目，利用同余定理依次求得所有 $k$ 倍区间数目。本人在洛谷提交的代码，注意 #3 的 $cnt$ 超出了范围，需要用 $long$ 来定义，但是 #4 到 #7 一直 RE，如下图，在各大网站搜索解决办法也都无济于事，最后在输入时就把 $s[i]$ 对 $k$ 取模才解决了这个 $bug$ 。

AC! 爽！