2017年蓝桥杯B组CC++ 第10题

xiaoxiao2021-02-28  114

K倍区间

 

N个数的数列A1,A2,...,An,定义S(i, j) = Ai+Ai+1+……+Aj(i<=j)。如果S(i, j)能被正整数k整除,那么称S(i, j)K倍区间。

测试数据共两行。第一行为两个正整数Nk,第二行为N个不同的正整数。

 

输入样例1

5 2

1 2 3 4 5

输入样例2

5 3

1 2 3 4 5

输出样例1

6

输出样例2

7

 

样例1解释:这6个区间分别为[2,2], [4,4], [1,3], [3, 5], [1, 4], [2, 5]

 

 

题解:

sum(i)表示A1+A2+……+Ai,那么S(i, j) = sum(j) - sum(i-1)K倍区间满足S(i, j) % K == 0,则[sum(j) - sum(i-1)] % K == 0,变换为sum(j) % K == sum(i-1) % K。可以边读入Ai边计算sum(i),同时做模K运算。然后,对于sum(R),假设有xsum(L)(L<R)的值与sum(R)相同,那么就有xk倍区间,累加到答案中。

 

参考代码:

#include<stdio.h> #include<string.h> #define maxn 100001 int cnt[maxn], a[maxn]; __int64 sum; int main() { int n, k; scanf("%d %d", &n, &k); int i; for(i = 1; i <= n; i++) { scanf("%d", &a[i]); } memset(cnt, 0, sizeof(cnt)); a[0] = 0; for(i = 1; i <= n; i++) { a[i] = (a[i] + a[i-1]) % k; } sum = 0; cnt[0] = 1; for(i = 1; i <= n; i++) { sum += (__int64)cnt[a[i]]; cnt[a[i]]++; } printf("%I64d\n", sum); return 0; }

转载请注明原文地址: https://www.6miu.com/read-34938.html

最新回复(0)