acwing 截断数组

smile6a6a

已于 2023-02-26 10:37:22 修改

阅读量111

点赞数

文章标签：算法 c++

于 2023-02-18 16:42:24 首次发布

本文链接：https://blog.csdn.net/m0_73832482/article/details/129101919

版权

截断数组

题目

给定一个长度为 n 的数组 a₁,a₂,…,a_n。

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

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

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

输入格式

第一行包含整数 n。

第二行包含 n 个整数 a₁,a₂,…,a_n。

输出格式

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

数据范围

前六个测试点满足 1≤**n≤10。
所有测试点满足 1≤n*≤10⁵，−10000≤a_i≤10000。

输入样例1：

4
1 2 3 3

输出样例1：

输入样例2：

5
1 2 3 4 5

输出样例2：

输入样例3：

2
0 0

输出样例3：

难度：中等
时/空限制：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;
}