Educational Codeforces Round 21 D. Array Division

题目链接：Educational Codeforces Round 21 D. Array Division

题意：

给你n个数，现在你可以改变1<=个数的位置，然后问你是否存在有一个k，使得sum(a[i])(1<=i<=k)==sum(a[j])(k+1<=j<=n)

题解：

 分析：

如果需要将一个数移动，无非就是将这个数从第一部分移到第二部分，或者从第二部分移到第一部分。

所以，我们只需要开两个map来记录一下两部分有哪些数。

当两部分的差值/2等于其中一部分的一个数时，那么就可以YES了。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long ll;

const int N=1e5+7;
ll a[N];
map<ll,int>mp;
map<ll,int>mp2;
int main()
{
    int n;ll sum1=0,sum2=0;
    scanf("%d",&n);
    F(i,1,n)scanf("%lld",a+i),sum2+=a[i],mp[a[i]]++;
    if(sum2&1){puts("NO");return 0;}
    ll aim=sum2/2;
    F(i,1,n)
    {
        ll tp=sum2-aim;
        if(tp>0&&mp[tp]>0){puts("YES");return 0;}
        if(tp<0&&mp2[-tp]>0){puts("YES");return 0;}
        sum2-=a[i];
        mp[a[i]]--;
        mp2[a[i]]++;
    }
    puts("NO");
    return 0;
}