问题:假设有一个数组x[],有n个元素,并且每一个都大于零;称x[0]+x[1]+x[2]+...+x[i]为前置和,而x[j]+x[j+1]+...+x[n-1]为后置和。试编写一个程序,求出x[]中有多少组前置和后置和。
思路:设置两个索引变脸indexHead和indexTail,一个从前往后扫描,一个从后往前扫描。并且使用两个变量记录当前的前置和和后置和,并通过比较他们的值来决定移动哪个索引变量。直接看代码吧。
#include <stdio.h>
int countHeadTailNum(const int a[], int n) {
int indexHead = 0,indexTail = n -1;
int sumHead = 0,sumTail = 0;
int nums = 0;
while(indexHead <= n - 1 && indexTail >= 0) {
if(sumHead > sumTail) {
sumTail += a[indexTail--];
}
else if(sumHead < sumTail) {
sumHead += a[indexHead++];
}
else {
nums++;
sumHead += a[indexHead++];
sumTail += a[indexTail--];
}
}
return nums;
}
int main() {
int a[] = {3,6,2,1,4,5,2};
printf("The number of equal prefix sum and sufix sum is %d\n",countHeadTailNum(a,7));
return 0;
}