3788 截断数组（前缀和）

1. 问题描述：

给定一个长度为 n 的数组 a1，a2，…，an。现在，要将该数组从中间截断，得到两个非空子数组。要求，第一个子数组内各元素之和等于第二个子数组内各元素之和。请问，共有多少种不同的截断方法？

输入格式

第一行包含整数 n。第二行包含 n 个整数 a1，a2，…，an。

输出格式

输出一个整数，表示截断方法数量。

数据范围

前三个测试点满足 1 ≤ n ≤ 10。
所有测试点满足 1 ≤ n ≤ 10 ^ 5，−10000 ≤ ai ≤ 10000。

输入样例1：

9
1 5 -6 7 9 -16 0 -2 2

输出样例1：

3

输入样例2：

3
1 1 1

输出样例2：

0

输入样例3：

4
100 1 10 111

输出样例3：

1
来源：https://www.acwing.com/problem/content/3791/

2. 思路分析：

分析题目可以知道我们可以枚举第一个位置到倒数第二个位置，判断当前切的位置是否满足题目的要求，也即需要判断前后两段数组的和是否相等，如何快速计算区间内所有数字的和呢？我们可以预处理出前缀和，然后枚举每一个可以切的位置，使用前缀和数组判断是否相等即可。

3. 代码如下：

class Solution:
    def process(self):
        n = int(input())
        a = list(map(int, input().split()))
        s = [0]
        for i in range(n):
            s.append(s[-1] + a[i])
        res = 0
        for i in range(1, n):
            if s[i] == s[n] - s[i]:
                res += 1
        return res


if __name__ == "__main__":
    print(Solution().process())