D. Ones and Twos

原題連結

題解

對於任意和為 sum 的子陣列 \(S_1\)，其可以變成 sum-2 的子陣列 \(S_2\)

code

#include<bits/stdc++.h>
using namespace std;

int a[100005];
int main()
{
    int t;
    cin>>t;

    while(t--)
    {
        int n,q;
        cin>>n>>q;

        set<int> pos;
        int sum=0;
        for(int i=1;i<=n;i++)
        {
            cin>>a[i];
            sum+=a[i];
            if(a[i]==1) pos.insert(i);
        }

        while(q--)
        {
            int op;
            cin>>op;

            if(op==1)
            {
                int s;
                cin>>s;

                int len=pos.size();
                //printf("sum1:  %d\n",sum);
                if(len==0)
                {
                    if(s%2==sum%2) puts("yes");
                    else puts("no");
                    continue;
                }
                //printf("maxs:%d  mins:%d  size:%d\n",*pos.rbegin(),*pos.begin(),len);
                int sum1=2*(*pos.rbegin()-len)+len-1;
                int sum2=2*(n-*pos.begin()+1)-len-1;


                //printf("sum2: %d\nsum3: %d\n",sum1,sum2);

                if(s%2==sum%2)
                {
                    if(s<=sum) puts("yes");
                    else puts("no");
                }
                else
                {
                    if(s<=max(sum1,sum2)) puts("yes");
                    else puts("no");
                }
            }
            else
            {
                int id,v;
                cin>>id>>v;

                if(v==1)
                {
                    if(a[id]==2)
                    {
                        pos.insert(id);
                        sum--;
                    }
                    a[id]=1;
                }
                else
                {
                    if(a[id]==1)
                    {
                        pos.erase(id);
                        sum++;
                    }
                    a[id]=2;
                }
            }
        }

    }
    return 0;
}