P5057 [CQOI2006]简单题 （线段树入门）

题目链接

线段树维护RSQ问题，tree[rt]表示rt代表的区间中 1的个数，

思路和普通的RSQ问题处理基本一致，有一个要注意的地方在于，更新懒标记tag的时候，如果翻转了偶数次，那么相当于不翻转，翻转了奇数次等价于翻转1次。

所以我们更新方法为：tag[rt]^=1 ,
当tag[rt]==1 ，说明子树区间需要翻转，否则不需要

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e5+7;
const int INF =0x3f3f3f3f;
//线段树维护RSQ
int tree[maxn*4],tag[maxn*4];
int n,m;
inline int lc(int &rt)
{
    return rt<<1;
}
inline int rc(int &rt)
{
    return rt<<1|1;
}
inline void pushup(int &rt)
{
    tree[rt]=tree[lc(rt)]+tree[rc(rt)];
}
/*void build(int rt,int l,int r)
{
    if(l == r)
    {
        tree[rt]=0;
    }
    else 
    {
        int mid=(l+r)>>1;
        build(lc(rt),l,mid);
        build(rc(rt),mid+1,r);
        pushup(rt);
    }
}*/
inline void pushdown(int rt,int l,int r)
{
    if(tag[rt])
    {
        int mid=(l+r)>>1;
        tree[lc(rt)]=mid-l+1-tree[lc(rt)];
        tag[lc(rt)]^=1;
        tree[rc(rt)]=r-mid-tree[rc(rt)];
        tag[rc(rt)]^=1;
        tag[rt]=0;
    }
}
inline void updata(int rt,int l,int r,int vl,int vr)//[vl,vr]区间翻转
{
    if(vr<l || vl>r) return;
    if(vl<=l && r<=vr) 
    {
        tree[rt]=r-l+1-tree[rt];
        tag[rt]^=1;
        return ;
    }
    int mid=(l+r)>>1;
    pushdown(rt,l,r);
    updata(lc(rt),l,mid,vl,vr);
    updata(rc(rt),mid+1,r,vl,vr);
    pushup(rt);
}
inline int query(int rt,int l,int r,int vl,int vr)//询问区间[vl,vr]中1的个数
{
    if(vr<l || vl>r) return 0;
    if(vl<=l && r<=vr) 
    {
        return tree[rt];
    }
    int mid=(l+r)>>1;
    pushdown(rt,l,r);
    return query(lc(rt),l,mid,vl,vr)+query(rc(rt),mid+1,r,vl,vr);
}
int main()
{
    scanf("%d %d",&n,&m);
    int f,x,y;
//    build(1,1,n); 默认是0 所以建树就可以省略了
    while(m--)
    {
        scanf("%d",&f);
        if(f == 1)
        {
            scanf("%d %d",&x,&y);
            updata(1,1,n,x,y);
        }
        else 
        {
            scanf("%d",&x);
            printf("%d\n",query(1,1,n,x,x));//不管是单点查询还是区间查询，统一用区间查询
        }
    }
    return 0;
}