蓝桥杯截断数组

最新推荐文章于 2024-10-02 23:40:03 发布

甚小磊

最新推荐文章于 2024-10-02 23:40:03 发布

阅读量126

点赞数

文章标签：算法 c++ 蓝桥杯 Powered by 金山文档

本文链接：https://blog.csdn.net/weixin_57011178/article/details/129770887

版权

这是一个关于数组处理的编程问题，要求将一个长度为n的数组截断成三个元素和相等的非空子数组。程序读入数组长度和元素，然后计算不同截断方法的数量。当数组长度小于3或所有元素和不能被3整除时，没有解决方案。否则，通过遍历数组找到分割点来计算可能的截断方式。

摘要由CSDN通过智能技术生成

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

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

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

请问，共有多少种不同的截断方法？

输入格式

第一行包含整数 n�。

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

输出格式

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

数据范围

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

所有测试点满足 1≤n≤1051≤�≤105，−10000≤ai≤10000−10000≤��≤10000。

输入样例1：

4
1 2 3 3

输出样例1：

输入样例2：

5
1 2 3 4 5

输出样例2：

输入样例3：

2
0 0

输出样例3：

#include<iostream>
#include<cstring>
using namespace std;
const int N = 100000+100;
int n;
int ans[N];
int num[N];
long long k = 0;//记录数量
int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++)
    {
        scanf("%d",&ans[i]);
        num[i]=num[i-1]+ans[i];
        //cout<< num[i]<<" ";
    }
    
    // int l = 1; int r = n-1;
    if(n<3 || num[n]%3!=0){
        cout<<"0\n";
        return 0;
    }
    // while(l<r){
    //     cout<<(num[n]-num[r])<<" "<<num[r]-num[l]<<" "<<num[l]<<"\n";
    //     if((num[n]-num[r])==(num[r]-num[l])&&(num[r]-num[l])==num[l]){
    //         k++;
    //     }
    //     l++;
    //     r--;
    // }
    
    for(int i = 1, cns = 0; i< n;i++){
        if(num[i]==num[n]/3*2){
            k +=cns;
        }
        if(num[i]==num[n]/3){
            cns++;
        }
    }
    cout<<k<<"\n";
    return 0;
}