bzoj1230[Usaco2008 Nov]lites 开关灯 线段树lazy标记

明明就是个裸的线段树还不让我一遍过。
tmd标记每次要取反而不能直接标记1或者0。。因为有可能传递下来的标记和修改的标记重叠了。。

#include
#include
#include
#include
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e5+5;
int a[N];
struct node
{
    int l,r,siz,lazy;
    int sum;
}t[N*20];
int n,m;
inline void build(int x,int l,int r)
{
    t[x].l=l;
    t[x].r=r;
    t[x].siz=r-l+1;
    t[x].lazy=0;
    if (l==r)return;
    int mid=(l+r)>>1;
    build(x<<1,l,mid);
    build(x<<1|1,mid+1,r);
}
inline void pushdown(int x)
{
    t[x].lazy=!t[x].lazy;
    t[x<<1].lazy=!t[x<<1].lazy;
    t[x<<1|1].lazy=!t[x<<1|1].lazy;
    //t[x<<1].lazy=t[x<<1|1].lazy=1;
    t[x<<1].sum=t[x<<1].siz-t[x<<1].sum;
    t[x<<1|1].sum=t[x<<1|1].siz-t[x<<1|1].sum;
}
inline void change(int x,int l,int r)
{
    if (t[x].l==l&&t[x].r==r)
    {
        t[x].sum=t[x].siz-t[x].sum;
        t[x].lazy=!t[x].lazy;
        return;
    }
    if (t[x].lazy&&t[x].l!=t[x].r)pushdown(x);
    int mid=(t[x].l+t[x].r)>>1;
    if (r<=mid)change(x<<1,l,r);
    else if (l>mid)change(x<<1|1,l,r);
    else change(x<<1,l,mid),change(x<<1|1,mid+1,r);
    t[x].sum=t[x<<1].sum+t[x<<1|1].sum;
}
inline int query(int x,int l,int r)
{

    if (t[x].l==l&&t[x].r==r)return t[x].sum;
    if (t[x].lazy&&t[x].l!=t[x].r)pushdown(x);
    int mid=(t[x].l+t[x].r)>>1;
    if (r<=mid)return query(x<<1,l,r);
    else if (l>mid)return query(x<<1|1,l,r);
    else return query(x<<1,l,mid)+query(x<<1|1,mid+1,r);
    //t[x].sum=t[x<<1].sum+t[x<<1|1].sum;
}
int main()
{
    scanf("%d%d",&n,&m);
    build(1,1,n);
    fo(i,1,m)
    {
        int l,r,op;
        scanf("%d%d%d",&op,&l,&r);
        if (op==0)
        {
            change(1,l,r);
        }
        else printf("%d\n",query(1,l,r));
    }
    return 0;
}

你可能感兴趣的:(bzoj,线段树)