题目:
小b喜欢和为K的倍数的序列。
现在有一个长度为n的序列A,请问A有多少个非空连续子序列是小b喜欢的。
输入
第一行输入一个正整数n; 第二行输入n个整数,表示A[i],以空格隔开; 第三行输入一个正整数K; 其中1≤n≤30000,对于任意A[i]有-10000≤A[i]≤10000,2≤K≤10000
输出
输出一个数,表示子序列的数目
输入样例
6
4 5 0 -2 -3 1
5
输出样例
7
题解:
暴力即可,注意要将数据范围控制在k内,%k即可。
代码:
#include<stdio.h>
#include<algorithm>
using namespace std;
int a[30010];
int sum[30010];
int main()
{
int n;
scanf("%d", &n);
for(int i = 1; i <= n; i++){
scanf("%d", &a[i]);
}
int k;
int ans = 0;
scanf("%d", &k);
for(int i = 1; i <= n; i++){
sum[i] = (sum[i - 1] + a[i]) % k;
}
for(int i = 0; i <= n; i++)
for(int j = i + 1; j <= n; j++){
if((sum[j] - sum[i]) % k == 0)
ans++;
}
printf("%d\n", ans);
}