K倍区间(前缀和+同余)

题目描述

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

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

输入格式

第一行包含两个整数 N 和 K。

以下 N 行每行包含一个整数 Ai。

5 2
1
2
3
4
5

输出格式

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

6

数据范围
$1\le N,K\le 100000$,
$1\le A_i\le 100000$

C++ (前缀和+同余)

#include <iostream>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
int n, k;
LL s[N], cnt[N];

int main() {
	scanf("%d%d", &n, &k);
	
	for (int i = 1; i <= n;i++) {
		scanf("%d", &s[i]);
		s[i] += s[i-1];
	}
	
	LL res = 0; 
	cnt[0] = 1; // // 第一次遇到区间和为0时，应该直接+1，s[i]=0指的是从1到i直接就是满足条件的
	for (int i = 1; i <= n; i++) {
		res += cnt[s[i]%k]; // 这个 s[i]%k 可以和前面取模为 s[i]%k的分别构成一个k倍区间
							// 所以此时又新增了cnt[s[i]%k]个K倍区间
		cnt[s[i]%k]++;
	}
	
	cout << res << endl;
	
	return 0;
}