3956. 截断数组

3956. 截断数组 - AcWing题库

 3956. 截断数组

【题目描述】

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

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

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

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

【输入】

第一行包含整数 nn。

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

【输入】

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

数据范围

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

解题思路：

因为题意是由一个不变的数组，截成三段，所以这个数组的总和 sum 是相等的，其中截成的三段的值要都相等，那么这三段应该截成的三段它们的和应该满足： sum1==sum2==sum3==sum/3 .首先想到的是用前缀和，后缀和，因为要判断的情况太多了，刚开始是这么写的：

大概是先判断前缀和到达 sum1==sum/3 的时候就判断后缀和，但是会有漏掉的情况，因为 i 层的循环是一直自增的，判断完第一段满足要求后，接着要遍历后一段满足要求的区域，此时可以用数组来存储后一段满足条件的 sum3==sum 的部分（数据过大时可能会重复计算很多遍），而且还要保证统计出来的数量没有重复的部分。

一直在改的错误代码：

#include<stdio.h>
int a[100005],book[100005];
int main(){
	int n,sum=0;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
		sum=sum+a[i];
	}
	int x=sum/3;
	if(sum%3!=0)
	{
		printf("0\n");
		return 0;
	}
	int S=0;
	int sum1=0,sum2=0;
	int k=n-1,i,j;
	for(i=0;i<n;i++){
		sum1=sum1+a[i];
		if(sum1==x){
			for(k=n-1;k>i+1;k--){
				if(sum2==x)
				{
					S++;
					book[k]=1;
				}
				sum2=sum2+a[k];
				n--;
				if(book[k]==0&&sum2==x)
				{
					S++;
					book[k]=0;
				}
			}
		}
	} 
	printf("%d\n",S);
	return 0;
} 

然后，看到题解，写的很简单。（sum 是数组的总和）

他的思路是记录前缀和（sum1）中满足 sum1==sum/3 的部分（也就是第一次截断的点）以及满足 sum1==sum/3*2 的部分（第二次截断的点）。

数据有些大，要开 long long 存储。

#include<stdio.h>
int a[100005];
int main(){
	int n,x,sum=0;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
		sum=sum+a[i];
	}
	x=sum/3;
	long long S=0,ans=0;
	long long flag=0;
	if(sum%3!=0||n<3){
		printf("0\n");
		return 0;
	}
	for(int i=0;i<n-1;i++){//第二次截断后，第三个位置不能为空 
		S=S+a[i];
		if(S==2*x)
		ans=ans+flag;
		if(S==x)
		flag++;
	}
	printf("%lld\n",ans);
	return 0; 
}

795.前缀和

代码如下：

#include<stdio.h>
int sum[100005];
int main(){
	int a,b,x,n,m;
	scanf("%d%d",&n,&m); 
	for(int i=1;i<=n;i++){
		scanf("%d",&x);
		sum[i]=sum[i-1]+x;
	} 
	for(int i=0;i<m;i++){
		scanf("%d%d",&a,&b);
		printf("%d\n",sum[b]-sum[a-1]);
	}
	return 0;
}

796.子矩阵的和

 代码如下：

#include<stdio.h>
int a[1005][1005],sum[1005][1005];
int main(){
	int x,y,z,w,i,j,n,m,k;
	scanf("%d%d%d",&n,&m,&k);
	for(i=1;i<=n;i++){
		for(j=1;j<=m;j++){
			scanf("%d",&a[i][j]);
		}
	}
	for(i=1;i<=n;i++){
		for(j=1;j<=m;j++){
			sum[i][j]=a[i][j]+sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1];
		}
	}
	
	for(i=0;i<k;i++){
		scanf("%d%d%d%d",&x,&y,&z,&w);
		printf("%d\n",sum[z][w]-sum[x-1][w]-sum[z][y-1]+sum[x-1][y-1]);
	}
}