P3870 [TJOI2009]开关（线段树模板题）

开两个sum变量分别维护开的灯数和关的灯数
开关的时候就交换一下

题目描述
现有N（2 ≤ N ≤ 100000）盏灯排成一排，从左到右依次编号为：1，2，…，N。然后依次执行M（1 ≤ M ≤ 100000）项操作，操作分为两种：第一种操作指定一个区间[a, b]，然后改变编号在这个区间内的灯的状态（把开着的灯关上，关着的灯打开），第二种操作是指定一个区间[a, b]，要求你输出这个区间内有多少盏灯是打开的。灯在初始时都是关着的。

输入输出格式
输入格式：
第一行有两个整数N和M，分别表示灯的数目和操作的数目。接下来有M行，每行有三个整数，依次为：c, a, b。其中c表示操作的种类，当c的值为0时，表示是第一种操作。当c的值为1时表示是第二种操作。a和b则分别表示了操作区间的左右边界（1 ≤ a ≤ b ≤ N）。

输出格式：
每当遇到第二种操作时，输出一行，包含一个整数：此时在查询的区间中打开的灯的数目。

输入输出样例
输入样例#1：
4 5
0 1 2
0 2 4
1 2 3
0 2 4
1 1 4
输出样例#1：
1
2

#include<iostream>
#include<stdio.h>
using namespace std;

    struct st{
        int l,r;
        long long sum0,sum1,add;
        #define l(x) tree[x].l
        #define r(x) tree[x].r
        #define sum0(x) tree[x].sum0
        #define sum1(x) tree[x].sum1
        #define add(x) tree[x].add
    }tree[100010*4];
    int a[100010],n,m;
    
    void build(int p,int l,int r){
        l(p)=l;r(p)=r;
        if(l==r){
            sum1(p)=0;
            sum0(p)=1;
// printf("%d %d %d\n",l,r,sum(p));
            return;
        }
        int mid=(l+r)/2;
        build(p*2,l,mid);
        build(p*2+1,mid+1,r);
        sum0(p)=sum0(p*2)+sum0(p*2+1);
        sum1(p)=sum1(p*2)+sum1(p*2+1);
    }
    
    void spread(int p){
        if(add(p)%2==1){
// sum(p*2)+=add(p)*(r(p*2)-l(p*2)+1);
// sum(p*2+1)+=add(p)*(r(p*2+1)-l(p*2+1)+1);
			swap(sum0(p*2),sum1(p*2));
			swap(sum1(p*2+1),sum0(p*2+1)); 
            add(p*2)++;
            add(p*2+1)++;
            add(p)=0;
        }
    }
    
    void change(int p,int l,int r,int d){
        if(l<=l(p)&&r>=r(p)){
            swap(sum0(p),sum1(p));
            add(p)++;
            return;
        }
        spread(p);
        int mid=(l(p)+r(p))/2;
        if(l<=mid) change(p*2,l,r,d);
        if(r>mid) change(p*2+1,l,r,d);
        sum1(p)=sum1(p*2)+sum1(p*2+1);
        sum0(p)=sum0(p*2)+sum0(p*2+1);
    }
    
    long long ask(int p,int l,int r){

        if(l<=l(p)&&r>=r(p)) return sum1(p);
        spread(p);
        int mid=(l(p)+r(p))/2;
        long long  val=0;
// printf("l:%d l(p):%d r:%d r(p):%d\n",l,l(p),r,r(p));
        // cout<<"#";
        if(l<=mid) val+=ask(p*2,l,r);
        if(r>mid) val+=ask(p*2+1,l,r);
        return val; 
    }
    
    int main(){
        int n,m;scanf("%d%d",&n,&m);
  // for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        build(1,1,n);
        for(int i=1;i<=m;i++){
            int t;scanf("%d",&t);
            if(t==0){
                int x,y,k;scanf("%d%d",&x,&y);
                change(1,x,y,k);
            } 
            else if(t==1){
                int x,y;scanf("%d%d",&x,&y);
                printf("%lld\n",ask(1,x,y));
            }
        }
    }