codeforces Educational Codeforces Round 21 D. Array Division

最新推荐文章于 2022-07-08 21:41:28 发布

原创最新推荐文章于 2022-07-08 21:41:28 发布 · 351 阅读

0 ·

CC 4.0 BY-SA版权

codeforces 同时被 2 个专栏收录

20 篇文章

订阅专栏

枚举

4 篇文章

订阅专栏

本文介绍了一种判断数组是否可以通过重新排列元素分为两部分，使得每部分元素之和相等的算法。通过使用前缀和与后缀和的概念，并借助哈希表记录元素的位置信息，该算法能在O(n)的时间复杂度内解决问题。

分两种情况，一个是后面的数放到前面，另一种是前面的数放到后面（其实还有一种是无变化，可以放到其中一种来看）

分别记录前缀和和后缀和，用两个map记录某个数出现的最早坐标和最晚坐标，首先如果总和是奇数就输出NO，否则就进行从前向后遍历，若出现最晚坐标的map[总和/2-前缀和]>当前坐标就输出YES，若出现总和/2==前缀和也输出YES，另外再从后向前遍历一遍，用后缀和和最早坐标的map来作相似操作即可，如果最后都没输出就输出NO。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<ctime>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<set>
#include<bitset>
#include<map>
#include<tr1/unordered_map>
#include<stack>
#include<queue>
#include<vector>
#include<utility>
#define INF 0x3f3f3f3f
#define inf 2*0x3f3f3f3f
#define llinf 1000000000000000000
#define pi acos(-1.0)
#define mod 1000000007
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define lrt rt<<1
#define rrt rt<<1|1
#define rep(i,a,b) for(int i=(a);i<(b);i++)
#define per(i,a,b) for(int i=(b)-1;i>=(a);i--)
#define mem(a,b) memset(a,b,sizeof(a))
#define lb(x) (x&-x)
#define gi(x) scanf("%d",&x)
#define gi2(x,y) scanf("%d%d",&x,&y)
#define gll(x) scanf("%lld",&x)
#define gll2(x,y) scanf("%lld%lld",&x,&y)
#define gc(x) scanf("%c",&x)
#define gc2(x,y) scanf("%c%c",&x,&y)
using namespace std;
using namespace std::tr1;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int>P;
/***********************************************/
ll n,a[100005],sum[100005],bsum[100005];
map<ll ,ll>ma,mi;
int main()
{
    gll(n);
    rep(i,1,n+1){gll(a[i]);sum[i]=sum[i-1]+a[i];ma[a[i]]=i;}
    per(i,1,n+1){bsum[i]=bsum[i+1]+a[i];mi[a[i]]=i;}
    if(sum[n]&1)return cout<<"NO",0;
    ll s=sum[n]/2;
    rep(i,0,n)
         if(s-sum[i]>0&&ma[s-sum[i]]>i)return cout<<"YES",0;
         else if(s-sum[i]==0)return cout<<"YES",0;
    per(i,2,n+2)if(mi[s-bsum[i]]!=0&&mi[s-bsum[i]]<i)return cout<<"YES",0;
    cout<<"NO";
    return 0;
}