题目:
小b喜欢和为K的倍数的序列。
现在有一个长度为n的序列A,请问A有多少个非空连续子序列是小b喜欢的。
输入
第一行输入一个正整数n;
第二行输入n个整数,表示A[i],以空格隔开;
第三行输入一个正整数K;
其中1≤n≤30000,对于任意A[i]有-10000≤A[i]≤10000,2≤K≤10000
输出
输出一个数,表示子序列的数目
输入样例
输出样例
题解:
暴力即可,注意要将数据范围控制在k内,%k即可。
代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
| #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); }
|