截断数组
题目
给定一个长度为 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
难度:中等 |
---|
时/空限制:1s / 256MB |
总通过数:7736 |
总尝试数:31751 |
来源:AcWing,第16场周赛 |
算法标签 |
题解:
这题的主要思路在于可截断为3个和相同的部分的必要条件。
就是数组总和为3的倍数,然后遍历即可。
#define _CRT_SECURE_NO_WARNINGS 1
/****************************************************/
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 1e5 + 10;
int sum[maxn];
int a[maxn];
int main()
{
int n;
cin >> n;
int mina = 0, maxa = 0;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
sum[i] = sum[i - 1] + a[i];
mina = min(mina, a[i]);
maxa = max(maxa, a[i]);
}
int x = sum[n] / 3;
vector<int>b;
if (sum[n] % 3 != 0)
{
printf("0");
return 0;
}
for (int i = 1; i <= n - 2; i++)
{
if (sum[i] == x)b.push_back(i);
}
if (sum[n] == 0 && mina == maxa)
{
printf("%lld", (long long)(n - 2) * (n - 1) / 2);
return 0;
}
long long ans = 0;
int m[maxn],m0=0;
for (int i = 1; i <= n-1; i++)
{
if (sum[i] == x + x)m[m0++] = i;
}
int h1=0, h2=0;
while (b.size())
{
int t = b.back();
b.pop_back();
int *y = upper_bound(m, m + m0 - 1, t) ;
int v = *y;
int c = y - m;
if (v > t)ans += m0 - c;
}
printf("%d", ans);
return 0;
}