cf808D(xjb)

题目链接：http://codeforces.com/problemset/problem/808/D

 

题意：问能不能通过交换不超过两个元素的位置使得原数组变成能分成前，后和相等的连续两部分；

注意这里只能交换一次！！！

 

思路：若存在某段前缀和 sum1[i] + x = ans/2 其中 x 为[ 0, i ] 外的某个元素 或者后缀和 sum2[i] + x = ans/2 其中 x 为 [ i, n-1 ] 外的某个元素；

则为 "YES" ，否则为 "NO" ；

 

代码：

#include <iostream>
#include <stdio.h>
#include <map>
#define ll long long
using namespace std;

const int MAXN = 1e5+10;
map<ll, int> mp1, mp2;
int a[MAXN];

int main(void){
    int n;
    ll ans=0;
    scanf("%d", &n);
    for(int i=0; i<n; i++){
        scanf("%d", &a[i]);
        ans += a[i];
        mp1[a[i]]++;
        mp2[a[i]]++;
    }
    if(ans&1){
        cout << "NO" << endl;
        return 0;
    }
    ans >>= 1;
    ll num1=0, num2=0;
    for(int i=0; i<n; i++){
        if(mp1[ans-num1] > 0){
            cout << "YES" << endl;
            return 0;
        }
        num1 += a[i];
        mp1[a[i]]--;
    }
   for(int i=n-1; i>=0; i--){
        if(mp2[ans-num2] > 0){
            cout << "YES" << endl;
            return 0;
        }
        num2 += a[i];
        mp2[a[i]]--;
    }
    cout << "NO" << endl;
    return 0;
}