XOR的艺术(线段树)

题目

传送门

题解

比线段树2还要简单;异或的操作

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn=1000000;
const int inf=1e9;

int n,m,a[maxn],sum[maxn],tag[maxn];
bool leaves[maxn];
char ch[maxn];

void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void pushdown(int rt,int tot)//tot子树的节点的数量 
{
    if (tag[rt])
    {
        tag[rt<<1]^=tag[rt];
        tag[rt<<1|1]^=tag[rt];
        sum[rt<<1]=(tot-(tot>>1))-sum[rt<<1];//异或的运算 
        sum[rt<<1|1]=(tot>>1)-sum[rt<<1|1];
        tag[rt]=0;
    }
}

void build_tree(int rt,int l,int r)
{
    if (l==r)
    {
        sum[rt]=a[l]; return;
    }
    int mid=(l+r)>>1;
    build_tree(rt<<1,l,mid);
    build_tree(rt<<1|1,mid+1,r);
    pushup(rt);
}

void change(int rt,int l,int r,int L,int R)
{
    if (L<=l && r<=R)
    {
        tag[rt]^=1;
        sum[rt]=r-l+1-sum[rt];
        return;
    }
    pushdown(rt,r-l+1);
    int mid=(l+r)>>1;
    if (L<=mid) change(rt<<1,l,mid,L,R);
    if (R>mid) change(rt<<1|1,mid+1,r,L,R);
    pushup(rt);
}

int ques(int rt,int l,int r,int L,int R)
{
    if (L<=l && r<=R) return sum[rt];
    pushdown(rt,r-l+1);
    int mid=(l+r)>>1; int ans=0;
    if (L<=mid) ans+=ques(rt<<1,l,mid,L,R);
    if (R>mid) ans+=ques(rt<<1|1,mid+1,r,L,R);
    return ans;
}

int main()
{
    scanf("%d%d",&n,&m);
    scanf("%s",ch+1);
    for (int i=1; i<=n; i++) a[i]=ch[i]-'0';
    build_tree(1,1,n);
    for (int i=1; i<=m; i++)
    {
        int opt,x,y; scanf("%d%d%d",&opt,&x,&y);
        if (!opt) change(1,1,n,x,y);
        else printf("%d\n",ques(1,1,n,x,y));
    }
    return 0;
}

总结

今天上午做了两道水题,都没有1A
我自己写错的地方在于tot数组搞得出了问题,异或在区间里就是总数减去sum值

猜你喜欢

转载自blog.csdn.net/A_Comme_Amour/article/details/79681126