ACWING/3816. 移动元素

前缀和 哈希集合

思路

首先求一个前缀和的存储数组long long sumn，方便后续操作

首先特殊情况：

当整个和sum[n]为奇数，那么一定无法分为对称两半，为false

普通情况：

• 从数组的第一个开始向后遍历，若当前的idnex前缀和大于了整体和的一半，则找一下前面已经遍历的元素是否存在一个元素恰好等于此差，存在就踢出去，此时满足分半的要求。这里剔除为了加快查找使用了哈希集合来找到这个元素

• 注意当前算法只解决了前面一部分数组可以剔除数字来达到目标，还可能是后一部分数字通过剔除来达到目标，所以反向再操作一次

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <unordered_set>

using namespace std;

const int N = 100010;
int n;
long long a[N],b[N]; // a为正向，b为逆向
long long sum[N];

bool check(long long num[]){
    // 前缀和
    for(int i = 1;i<=n;i++) sum[i] = sum[i-1] + num[i];
    // 判断如果是奇数则一定为false
    if(sum[n] % 2) return false;
    unordered_set<long long> st;
    for(int i = 1;i<=n;i++){
        st.insert(num[i]);
        if(st.count(sum[i] - sum[n]/2)) {
        // 若存在当前前缀和大于半和，且能在前面已有的元素找到剔除
            return true;
        }
    }
    return false;
}

int main(){
    int T;
    cin>>T;
    memset(sum,0,sizeof(sum));
    while(T--){
        cin>>n;
        for(int i = 1,j = n;i<=n;i++,j--){
            cin>>a[i];
            b[j] = a[i]; // a为正向，b为逆向
        }
        if(check(a) || check(b)) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
}