Acwing3956截断数组---枚举、前缀和

3956截断数组

给定一个长度为 n 的数组 a1,a2,…,an。
现在，要将该数组从中间截断，得到三个非空子数组。
要求，三个子数组内各元素之和都相等。
请问，共有多少种不同的截断方法？
输入格式
第一行包含整数 n。
第二行包含 n
个整数 a1,a2,…,an。
输出格式
输出一个整数，表示截断方法数量。
数据范围
前六个测试点满足 1≤n≤10。
所有测试点满足 1≤n≤105，−10000≤ai≤10000。
输入样例1：

4
1 2 3 3

输出样例1：

1

输入样例2：

5
1 2 3 4 5

输出样例2：

0

输入样例3：

2
0 0

输出样例3：

0

思路

首先计算出前缀和，分成三个和相等的区间也就是枚举两个点，使得数组分成三份，但是暴力枚举两个点会超时，所以要优化，只枚举其中的一个点。
枚举第二刀i，判断i是否符合题目要求，i到n是前缀和数组的第三份，判断第三份是否==num[n]/3。如果满足，那么就去看在i前面有多少个满足题目要求的就可以了。

import java.util.*;
public class Main {

  public static void main(String[] args) {
  	// TODO Auto-generated method stub
  	Scanner scanner = new Scanner(System.in);
  	int n = scanner.nextInt();
  	long[] num = new long[100010];
  	// 直接暴力枚举区间会超时 所以枚举一个点
  	for (int i = 1; i <= n; i++) { // 计算前缀和
  		int a = scanner.nextInt();
  		num[i] = num[i - 1] + a;
  	}
  	long res = 0; // 保存答案 最后返回
  	// 判断如果能被整分成三分 那么也一定是3的倍数 如果不是3的倍数 就返回0
  	if (num[n] % 3 != 0) {
  		System.out.println("0");
  		return;
  	}
  	long count = 0; // 保存在i点之前有多少个可以满足 num[n]/3的点
  	for (int i = 3; i <= n; i++) {
  		//
  		if (num[i - 2] == num[n] / 3) { // 第二刀前面有多少个满足要求的点
  			// 这个count一直保存不会因为for循环而清零 所以他是保存的枚举到第i点的时候
  			// i点前面一共有多少个可以满足要求的点
  			// i点也是从3开始枚举 因为如果想要分成三分最少三个点
  			count++;
  		}
  		if (num[n] - num[i - 1] == num[n] / 3) { // 这是枚举第二刀 判断第二刀是否满足要求
  			res = res + count;
  		}
  	}
  	System.out.println(res);

  }
}