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

## 样例 #1

### 样例输入 #1

```
5 2
1  
2  
3  
4  
5
```

### 样例输出 #1

```
6
```

连续子序列和首先想到前缀和，满足要求的和为 (s[r] - s[l]) % k == 0， 如果双层循环遍历的话肯定会超时，所以利用一个特定 如果 (s[r] - s[l]) % k == 0， s[r]%k == s[l]%k，这样就可以遍历每一个前缀和，然后求出相同余数的有多少个存到cnt数组中，最后遍历一次cnt数组，因为余数相同就可以满足条件，所以可以任意从cnt[i]中选取两个数组合，利用组合数原理Cn2即可。

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

题目描述

给定一个长度为 N 的数列，A_1,A_2,  A_N，如果其中一段连续的子序列之和是 K 的倍数，我们就称这个区间 $[i,j]$ 是 $K$ 倍区间。

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

## 输入格式

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

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

## 输出格式

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

## 样例 #1

### 样例输入 #1

```
5 2
1  
2  
3  
4  
5
```

### 样例输出 #1

```
6
```

## 提示

时限 2 秒, 256M。蓝桥杯 2017 年第八届
*/
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const int N = 1e5 + 10;

int s[N], cnt[N];
int n, k;

int main(void)
{
    cin >> n >> k;

    cnt[0] = 1; //什么都不选也是一种
    ll res = 0;
    for(int i=1 ; i<=n ; i++)
    {
        int x;
        cin >> x;
        s[i] += s[i-1] + x;
        cnt[ s[i] % k ] ++;
    }

    for(int i=0 ; i<k ; i++)
    {
        res += 1ll * cnt[i] * (cnt[i] - 1) / 2;
    }

    cout << res << endl;
    return 0;
}