F - Triangles

最新推荐文章于 2025-02-07 10:06:54 发布

weixin_34259232

最新推荐文章于 2025-02-07 10:06:54 发布

阅读量203

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/shengge-777/p/9470368.html

本文介绍了一个使用C++实现的算法，该算法能够判断一个整数数组是否可以被等分为三个部分，且每个部分的总和相等。通过运用map来记录前缀和出现的次数，此算法能在O(n)的时间复杂度内完成任务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
map<int,int>vis;
int main()
{
int n,m,a[100001]={0},j,temp=0,i,k,pre[100001]={0};
cin>>n;
int sum=0;
for(i=1;i<=n;i++)
{
    cin>>a[i];
    pre[i]=a[i]+pre[i-1];
    vis[pre[i]]++;
    sum+=a[i];
}
if(sum%3!=0)
    cout<<0<<endl;
else
{
    int ans=sum/3;
    int count=0,temp=0,cnt=0;
 for(i=1;i<=n;i++)
 {
         if(vis[ans+pre[i]]&&vis[ans*2+pre[i]]) cnt++;
}
 cout<<cnt<<endl;
}

return 0;
}