题:截断数组

最新推荐文章于 2024-07-14 22:18:32 发布

duolade_Ameng

最新推荐文章于 2024-07-14 22:18:32 发布

阅读量123

点赞数

分类专栏：算法题文章标签： java 算法数据结构

本文链接：https://blog.csdn.net/weixin_60470198/article/details/129019466

版权

算法题专栏收录该内容

5 篇文章 0 订阅

订阅专栏

🎈描述

https://www.acwing.com/problem/content/3959/
给定一个长度为 n的数组 a1,a2,…,an。
现在，要将该数组从中间截断，得到三个非空子数组。
要求，三个子数组内各元素之和都相等。
请问，共有多少种不同的截断方法？

输入格式
第一行包含整数 n。
第二行包含 n个整数 a1,a2,…,an。

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

数据范围

前六个测试点满足 1≤n≤10。
所有测试点满足 $1≤n≤10^5，−10000≤a_i≤10000$ 。

💡思路

要用前缀和,这是显然的;
由数据量推断复杂度控制在nlogn 想到枚举一个分割点,二分另一个分割点,但题目中数组元素有负数,不满足二段性,也不行;
不会了,看答案😂

📝代码

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
	static BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
	static int n;
	static int N=100000;
	static long pre[]=new long[N+5];
	public static void main(String[] args) throws IOException {
		n=Integer.parseInt(br.readLine());
		String s[]=br.readLine().split(" ");
		for(int i=1;i<=n;i++) {
			pre[i]=Integer.parseInt(s[i-1])+pre[i-1];
		}
		if(pre[n]%3!=0) {
			System.out.println(0);
			return;
		}
		long cnt=0;
		long ans=0;
		long t=pre[n]/3;
		for(int j=2;j<n;j++) {
			if(pre[j-1]==t) cnt++;//统计[1,j-1]中有多少元素满足[1,该元素]区间和=sum/3
			if(pre[j]==t*2) ans+=cnt;
		}
		System.out.println(ans);
	}

}