【洛谷】P8649 [蓝桥杯 2017 省 B] k 倍区间 题解

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

题目链接

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$ 倍区间的数目。

样例 #1

样例输入 #1

5 2
1  
2  
3  
4  
5

样例输出 #1

6

提示

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

题解思路

因为提到了连续子序列之和，这里我们采用前缀和数组预处理，可以较好地降低时间复杂度。对于输入的原数组 $nums[n]$ (下标从1开始)，处理为前缀和数组即 $nums[i]+=nums[i-1]$ ，则从 $l-r$ 的子序列之和即 $nums[r]-nums[l-1]$ 。

题目需要 $nums[r]-nums[l-1]$ 能够整除 $k$ 。根据数论中的同余定理，$k|(nums[r]-nums[l-1])$ ，$nums[r]$ 与 $nums[l-1]$ 同余，即 $nums\left[r\right]\equiv nums\left[l-1\right]\left(modk\right)$。
所求即转换成，nums的前缀和数组值对 $k$ 的余数相等的情况进行组合。
其中，当余数为 $0$ 的时候，如果出现了 $x$ 次，那么由于题目中 $i\le j$ ,所以余数为 $0$ 时有 $C_x^2+x$ 种情况。
$$C_x^2+x=\frac{x\left(x-1\right)}{2}+\frac{2x}{2}=\frac{x\left(x+1\right)}{2}=C_{x+1}^2$$
所以余数为 $0$ 的计数需要提前 +1.

题解代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 100003;
ll n, k, nums[N], cal[N];

int main()
{
    scanf("%lld%lld", &n, &k);
    nums[0] = 0;
    for (register int i(1); i <= n; ++i)
    {
        scanf("%lld", &nums[i]);
    }
    for (register int i(1); i <= n; ++i)
    {
        nums[i] = nums[i - 1] + nums[i];
    }
    cal[0] = 1;
    for (register int i(1); i <= n; ++i)
    {
        cal[nums[i] % k]++;
    }
    ll sum = 0;
    for (register int i(0); i <= k; ++i)
    {
        sum += cal[i] * (cal[i] - 1) / 2;
    }
    printf("%lld", sum);
    return 0;
}