《蓝桥杯每日一题》 前缀和·Acwing 3956. 截断数组

简介: 《蓝桥杯每日一题》 前缀和·Acwing 3956. 截断数组

1.题目

https://www.acwing.com/problem/content/3959/

给定一个长度为 n 的数组a1,a2,…,an。

现在,要将该数组从中间截断,得到三个非空子数组。

要求,三个子数组内各元素之和都相等。


请问,共有多少种不同的截断方法?


输入格式

第一行包含整数 n。


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


输出格式

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


数据范围

前六个测试点满足 1≤n≤10。


所有测试点满足 1≤n≤10的五次方,−10000≤ai≤10000。


输入样例1:

1. 4
2. 1 2 3 3

输出样例1:

1

输入样例2:

1. 5
2. 1 2 3 4 5

输出样例2:

0

输入样例3:

1. 2
2. 0 0

输出样例3:

0


2.思路分析


既然要分配三个非空子数组,且值相同,那么不存在元素个数小于3,不存在原数组总和%3!=0的情况

pres[]为前缀和数组,perSum=原数组总和/3


那么我们只需要找到一个位置 pres[i]==perSum*2


此时res+=(pres[1~i-1]==averge)即可


因为第i位 满足2*perSum,pres[i+1~n]==perSum


所以只需要加上能使前i位的数组 分为 两个值为averge的位置 的数量即可


3.Ac代码


import java.io.*;
public class Main {
    static int N=100010;
    public static void main(String[] args) throws IOException {
        int n;
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        n=Integer.parseInt(br.readLine());
        String []s=br.readLine().split(" ");
        int []arr=new int[N];
        //初始化初始数组
        for (int i = 1; i <=n; i++)   arr[i]=Integer.parseInt(s[i-1]);
        int []pres=new int[N];
        //求前缀和数组
        for (int i = 1; i <=n; i++)   pres[i]=pres[i-1]+arr[i];
        //如果个数小于3,或者前缀和不是3的倍数,那么无法截断
        if(n<3 ||pres[n]%3!=0){
            System.out.println("0");
            return;
        }
        long perSum=pres[n]/3,res=0,fir = 0,t=0;
        //要给最后一个元素留下了,给第三个截断用
        for (int i = 1; i <n; i++) {
            t=pres[i];
            if(t==perSum*2)  res+=fir;
            if(t==perSum)  fir++;
            //不能调换这两个if判断的原因是,如果当前t等于perSum和perSum*2时(比如0特判等),就会出现在同一位自加
            //一旦出现自加,那么可以理解为中间那个截断元素个数为0,这是不可能的
        }
        System.out.println(res);
    }
}


相关文章
|
4月前
|
机器学习/深度学习 Java BI
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-970 数组移动
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-970 数组移动
49 0
|
4月前
|
算法 Java C语言
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-49 算法训练 寻找数组中最大值
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-49 算法训练 寻找数组中最大值
48 0
|
3月前
蓝桥杯动态规划-第五弹 最大子数组和 买卖股票最佳时机IV 第N个泰波那契数 环形数组
蓝桥杯动态规划-第五弹 最大子数组和 买卖股票最佳时机IV 第N个泰波那契数 环形数组
|
3月前
|
Java
2022蓝桥杯大赛软件类省赛Java大学B组G题 数组切分
2022蓝桥杯大赛软件类省赛Java大学B组G题 数组切分
21 0
|
4月前
|
算法 Java C语言
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-493 合并排序数组
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-493 合并排序数组
37 0
|
4月前
|
算法 Java C语言
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-79 删除数组零元素
第十四届蓝桥杯集训——练习解题阶段(无序阶段)-ALGO-79 删除数组零元素
34 0
|
4月前
|
存储 Java 索引
第十四届蓝桥杯集训——数组(一维)
第十四届蓝桥杯集训——数组(一维)
61 0
|
4月前
|
人工智能 算法 Java
数组元素的目标和(蓝桥杯每日一题)
数组元素的目标和(蓝桥杯每日一题)
36 0
|
4月前
|
人工智能 算法 Java
截断数组(蓝桥杯每日一题)
截断数组(蓝桥杯每日一题)
36 0
|
4月前
蓝桥杯vip测试题系统-数组求和(解题思路以及解题代码,手画思路图虽然丑丑的)
蓝桥杯vip测试题系统-数组求和(解题思路以及解题代码,手画思路图虽然丑丑的)
49 0