c语言写代码:夏老师有一个包含 n个正整数的数组 a,他想知道数组 a中有多少个子段的和是3的倍数。 数组的子段为数组中单个或若干个连续的元素,如序列 [1],[2,3],[2,3,4]均为数组 [1,2,3,4,5]的子段。 当且仅当两个子段的起止下标不同时,两个子段为不同的子段。 输入数据 第一行一个整数n。 第二行n个整数,用单个空格隔开,表示数组a中的各个元素。 输出数据 一个整数,数组 a 中有多少个子段的和是3的倍数。 输入样例 5 3 1 3 2 1 输出样例 4
时间: 2023-05-19 16:01:13 浏览: 90
这是一个典型的前缀和问题,我们可以用一个数组 preSum 来记录前缀和,即 preSum[i] 表示前 i 个数的和。那么对于一个子段 [l, r] 的和,可以表示为 preSum[r] - preSum[l-1]。因此,我们可以枚举子段的左右端点 l 和 r,然后计算 preSum[r] - preSum[l-1] 是否是 3 的倍数,如果是,则计数器加一。
具体实现如下:
```c
#include <stdio.h>
int main() {
int n;
scanf("%d", &n);
int a[n+1], preSum[n+1];
for (int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
preSum[i] = preSum[i-1] + a[i];
}
int cnt = 0;
for (int i = 1; i <= n; i++) {
for (int j = i; j <= n; j++) {
if ((preSum[j] - preSum[i-1]) % 3 == 0) {
cnt++;
}
}
}
printf("%d\n", cnt);
return 0;
}
```
对于输入样例,输出为 4。