E n^2 可过...

简单优化一下即可...,幽默数据

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n;
ll a[214514];
ll s[214514],ss[214514];
ll cnt;
ll ans;

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
  {
  	cin>>a[i];
  	s[i]=s[i-1]+a[i];
  	ss[i]=ss[i-1]+(a[i]>0);
	}
	
	cnt=s[n]/3;
	if(cnt*3!=s[n])
	{
		cout<<0;
		return 0;
	}

	for(int i=1;i<=n;i++)
	{
		if(s[i]==cnt&&ss[i]>0)
		{
			for(int j=i+1;j<=n;j++)
			  if(ss[j]-ss[i]>0&&ss[n]-ss[j]>0)
			  	if(s[j]-s[i]==cnt)
			  		ans++;
		}
	}		
	
	cout<<ans;  		  
				
	return 0;
}

全部评论
确实幽默,如果我有一个长度为2e5的1 -1重复,ans是组合数C(1e5-1,2)。
点赞 回复 分享
发布于 2024-11-25 08:22 河南
会不会是能满足s[i]==cnt的个数应该是log级别的
点赞 回复 分享
发布于 2024-11-24 21:18 河南

相关推荐

码农索隆:这种hr,建议全中国推广
点赞 评论 收藏
分享
05-26 09:07
已编辑
门头沟学院 Java
点赞 评论 收藏
分享
06-15 18:44
黄淮学院 Java
Lynn012:如果是居民楼还是算了吧,看着有点野呢
点赞 评论 收藏
分享
评论
4
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务