题目描述
给出一个包含N个整数的序列A,定义这个序列A的前缀和数组为SUM数组 ,当SUM数组中的第i个元素等于在i前面的三个元素的和,那么第i个元素就称为GOOD。 那么这个SUM数组中包含多少个GOOD 元素?
输入
第一行整数T表示数据组数(1<=T<=10)
每组数据以下格式:
输入的第一行包含一个整数N (1<=N<=100000), 表示序列A的长度。
输入的第二行包含N个用空格隔开的整数,表示序列A (-100000<=Ai<=100000).
输出
每组数据输出仅一行,输出这个SUM数组中包含多少个GOOD元素。
样例输入
3
4
1 1 1 3
4
1 2 3 10
6
5 -2 -3 1 3 3
样例输出
1
0
1
解题思路:
首先求出前缀和数组,b[i]即为当前值加上之前所有的值的和,例如:1 1 1 3的前缀和数组为1,2,3,6;1,2,3,10的前缀和数组为1,3,6,16,当这个数组中的第i个元素等于在i前面的三个元素的和时,t++,输出t。
代码如下:
#include<stdio.h>
#include<string.h>
int main()
{
int a,b,i,q,c[100010],t;
scanf("%d",&a);
while(a--)
{
scanf("%d",&b);
t=0;
memset(c,0,sizeof(c));
for(q=i=0;i<b;i++)
{
scanf("%d",&c[i]);
q+=c[i];
c[i]=q;
}
for(i=3;i<b;i++)
{
if(c[i]==c[i-1]+c[i-2]+c[i-3])
{
t++;
}
}
printf("%d\n",t);
}
return 0;
}