CF1491A K-th Largest Value 题解

原题链接
洛谷链接

题目大意

给出一个长度为 $n$ 的01序列，并执行 $m$ 次操作。操作有如下两种

1. 将 $a_x$ 取反（0变1，1变0）
2. 询问该序列中第 $k$ 大的数

解题思路

刚看到题的时候以为是树状数组QwQ，其实并不需要

在读入时，我们记录一下序列中 $1$ 的个数 $tot$
每次修改的时候，如果 $a_x=0$ 则让 $tot$ 加一，否则让 $tot$ 减一
最后在询问时，如果 $tot\ge k$，则答案为 $1$，否则为 $0$

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int Maxn=1e5+10;
int n,m,tot;
int a[Maxn];;
inline int read()
{
    int s=0,w=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0' && ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
    return s*w;
}
int main()
{
    // freopen("in.txt","r",stdin);
    n=read(),m=read();
    for(int i=1;i<=n;++i)
    a[i]=read(),tot+=a[i];
    while(m--)
    {
        int opt=read(),x=read();
        if(opt==1)
        {
            a[x]^=1; //将 a[x] 取反
            if(a[x])++tot;
            else --tot;
        }
        else
        {
            if(tot>=x)puts("1");
            else puts("0");
        }
    }
    return 0;
}