CodeTON Round 7 (Div. 1 + Div. 2, Rated, Prizes!) ：D：Ones and Twos

CodeTON Round 7 (Div. 1 + Div. 2, Rated, Prizes!) ：D：Ones and Twos
解题思路：
    首先，我们注意到一个非常特殊的数组是只有一和二的，这就证明了这道题的特殊性。接下来看题目给出的条件，说有两个操作，一个查询数字，一个更改数字。如果一个数组为1 2 1 2 1，要求查询的是5，那么，这个五可以是前后减掉两个一得到，如果数组是2 1 1 1 2 ,那么直接减掉2就能得到。
    我们可以发现，只要sum(数组数字总和)和x(查询数字)的奇偶性是一样的，那么，sum只要>=x就能通过减掉偶数数字去得到想要的x，因为减掉左右两边（左边可减掉0/1个数字，右边同理）的数字之后，还是满足能够剪掉左右两边的偶数数字。但是，sum<x就不可以了。
    如果说sum和x的奇偶性不一样，那么sum可以通过减去奇数个数字来去和x匹配，那么就需要有1的存在，如果1不存在，那么可以说明一定查询不到。如果数字1存在，那么，我们就应该贪心的想如何让减去奇数之后的这个sum最大（也就是删掉几个数字之后sum最大），那么就可以想，在左边或者右边减去，就能够贪心的得到最大的数字，找到最左边的1、最右边的1的位置，（a,b），那么减完之后的sum为max(sum-(a*2-1),sum-(b*2-1));
    那么，为什么会得到这样的式子呢，我们可以想到，既然是最左边和最右边的1，那么，他们还没找到之前的数字都是2，也就是2 2 2 2 …… 1,反过来看就是1+2*(n-1)也就是一个a1=1,d=2的等差数列。
    那么，之后就是奇偶性相同(sum & 1 == x & 1)，和上同理。

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
set<int> s[3];
void solve(){
    int n,q;
    
    cin >> n >> q;//输入
    int sum = 0;
    int a[n+1];
    for(int i = 1;i <= n;i ++) {
        cin >> a[i];
        sum+=a[i];//得到数组总和
        s[a[i]].insert(i);//得到1和2的所在的位置
    }
    int op,x,v;
    while(q--){
        cin >> op;
        if(op == 1){//第一种操作。查询
            cin >> x;
            if((sum & 1) == (x & 1)){//同奇偶性
                if(sum >=x) cout <<"YES\n";//比大小
                else cout << "NO\n";
            }
            else{
                if(!s[1].size()) cout << "NO\n";//没1
                else{
                    int mx = max(sum-((*s[1].begin())*2-1),sum-((n-(*s[1].rbegin())+1)*2-1));//begin最左边1的位置，rbegin最右边1的位置
                    if(mx >=x){
                        cout << "YES\n";//和奇偶性相同同理
                    }
                    else{
                        cout << "NO\n";
                    }
                }
            }
        }
        else{
            cin >> x >> v;//2操作，改变数字
            sum-=a[x];
            s[a[x]].erase(x);//抹去他的位置存在
            a[x]=v;
            sum+=a[x];
            s[a[x]].insert(x);//加入新的位置存在
        }
    }
    s[1].clear();//全局变量，要清除
    s[2].clear();
}
signed main(){
	int t;
	cin >> t;
	while(t--) {
		solve();
	}
	return 0;
}