codeforces 808D - Array Division

题意：一个数组中可以将一些元素的位置移动到任意位置，问能否将一个移动后数组从某一位置处分成两部分，使得这两部分和相等

思路：首先求出数组所有元素总和以及前缀和，总和不能与2整除直接判断NO，然后再枚举要移动的元素，这个元素可以可以往前移，也可以往后移。如果不移动元素，就存在两部分和相等，那么必有某个前缀和pre[x]=sum/2。现在考虑移动第I个元素，如果往前移动则存在某个小于I的x,pre[x]=sum/2 - a[I]  .如果将第I个元素往后移，存在某个大于I的x,pre[x]=sum/2 + a[I];

直观的理解是

1.往前移，存在某个x，pre[x]=sum/2-a[I],(x<I) .存在某个前缀和等于总和一半减去某个a[I].如果把移过来a[I]加上这个前缀和那么就等于总和的一半。

2.往后移，存在某个x，pre[x]=sum/2+a[I],(x>I),存在某个前缀和等于总和一半加上某个a[I],如果把这个“多余”的a[I]移到后面，那么就等于总和的一半

还有这种操作。。

#include<cstdio>
#define ll long long 
const int maxn = 100010;
ll pre[maxn];
int a[maxn];
bool check(ll t,int l,int r)
{
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(pre[mid]==t)return true;
		if(pre[mid]>t)
		r=mid-1;
		else if(pre[mid]<t)
		l=mid+1;
	}
	return false;
}
int main()
{
	int n;
	while(~scanf("%d",&n))
	{
		pre[0]=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			pre[i]=pre[i-1]+a[i];
		}
		if(pre[n]%2)
		{
			printf("NO\n");
			continue;
		}
		ll sum=pre[n]/2;
		int flag=0;
		for(int i=1;i<=n;i++)
		{
			if(check(sum-a[i],1,i-1)||check(sum+a[i],i+1,n))
			{
				flag=1;
			}
		}
		if(flag==1)printf("YES\n");
		else printf("NO\n");
	}
}