算法|Acwing1230. K倍区间算法|c++|前缀和|蓝桥杯

给定一个长度为 N 的数列，A1,A2,…AN，如果其中一段连续的子序列 Ai,Ai+1,…Aj 之和是 KK 的倍数，我们就称这个区间 [i,j] 是 K 倍区间。
你能求出数列中总共有多少个 K 倍区间吗？
输入格式
第一行包含两个整数 N和 K。
以下 N 行每行包含一个整数 Ai。
输出格式
输出一个整数，代表 K 倍区间的数目。
数据范围
1≤N,K≤100000,
1≤Ai≤100000
输入样例：

5 2 1 2 3 4 5

输出样例：

6

#include #include #include using namespace std; typedef long long LL; const int N = 100010; LL num[N] , yus[N]; int main() { int n , k ; LL ans = 0; scanf("%d%d", &n, &k); for (int i = 1; i <= n; i ++ ){ LL x; cin >> x; num[i] = num[i - 1] + x; } LL res = 0; yus[0] ++; for (int i = 1; i <= n; i ++ ){ ans += yus[num[i] % k]; yus[num[i] % k] ++; } cout << ans << endl; return 0; }

【算法|Acwing1230. K倍区间】