Codevs 1690 开关灯 USACO

1690 开关灯 USACO
时间限制: 1 s
空间限制: 128000 KB
题目等级 : 钻石 Diamond
传送门
题目描述 Description
YYX家门前的街上有N(2<=N<=100000)盏路灯，在晚上六点之前，这些路灯全是关着的，六点之后，会有M(2<=m<=100000)个人陆续按下开关，这些开关可以改变从第i盏灯到第j盏灯的状态，现在YYX想知道，从第x盏灯到第y盏灯中有多少是亮着的(1<=i,j,x,y<=N)
输入描述 Input Description
第 1 行: 用空格隔开的两个整数N和M
第 2..M+1 行: 每行表示一个操作, 有三个用空格分开的整数: 指令号(0代表按下开关，1代表询问状态), x 和 y
输出描述 Output Description
第 1..询问总次数 行:对于每一次询问，输出询问的结果
样例输入 Sample Input
4 5
0 1 2
0 2 4
1 2 3
0 2 4
1 1 4
样例输出 Sample Output
1
2
数据范围及提示 Data Size & Hint
一共4盏灯，5个操作，下面是每次操作的状态(X代表关上的，O代表开着的)：
XXXX -> OOXX -> OXOO -> 询问1~3 -> OOXX -> 询问1~4
分类标签 Tags
线段树 树结构 USACO

/*
区间修改+区间求和+异或运算. 
*/
#include
#include
#include
#define LL long long
#define MAXN 100001
using namespace std;
int n,m,tot;
struct data{
    int l,r,lc,rc,sum,bj;
}tree[MAXN*4];
int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=x*10+ch-48,ch=getchar();
    return x*f;
}
void build(int l,int r){
    int k=++tot;
    tree[k].l=l,tree[k].r=r;
    if(l==r){
        tree[k].sum=0;
        return ;
    }
    int mid=(l+r)>>1;
    tree[k].lc=tot+1;
    build(l,mid);
    tree[k].rc=tot+1;
    build(mid+1,r);
    tree[k].sum=tree[tree[k].lc].sum+tree[tree[k].rc].sum;
}
void pushdown(int k){
    tree[k].bj=!tree[k].bj;
    tree[tree[k].lc].bj=!tree[tree[k].lc].bj;
    tree[tree[k].rc].bj=!tree[tree[k].rc].bj;
    tree[tree[k].lc].sum=abs(tree[tree[k].lc].r-tree[tree[k].lc].l+1-tree[tree[k].lc].sum);
    tree[tree[k].rc].sum=abs(tree[tree[k].rc].r-tree[tree[k].rc].l+1-tree[tree[k].rc].sum);
}
void change(int k,int l,int r){
    if(l<=tree[k].l&&tree[k].r<=r){
        tree[k].sum=abs(tree[k].r-tree[k].l+1-tree[k].sum);
        tree[k].bj=!tree[k].bj;
        return ;
    }
    if(tree[k].bj) pushdown(k);
    int mid=(tree[k].l+tree[k].r)>>1;
    if(r<=mid) change(tree[k].lc,l,r);
    else if(l>mid) change(tree[k].rc,l,r);
    else change(tree[k].lc,l,mid),change(tree[k].rc,mid+1,r);
    tree[k].sum=tree[tree[k].lc].sum+tree[tree[k].rc].sum;
}
int query(int k,int l,int r){
    if(l<=tree[k].l&&tree[k].r<=r) return tree[k].sum;
    if(tree[k].bj) pushdown(k);
    LL tot=0;
    int mid=(tree[k].l+tree[k].r)>>1;
    if(r<=mid) tot+=query(tree[k].lc,l,r);
    else if(l>mid) tot+=query(tree[k].rc,l,r);
    else tot+=query(tree[k].lc,l,mid),tot+=query(tree[k].rc,mid+1,r);
    tree[k].sum=tree[tree[k].lc].sum+tree[tree[k].rc].sum;
    return tot;
}
int main(){
    int x,y,z;
    n=read();
    build(1,n);
    m=read();
    for(int i=1;i<=m;i++){
        z=read();x=read();y=read();
        if(!z) change(1,x,y);
        else printf("%lld\n",query(1,x,y));
    }
    return 0;
}