CODEVS1690|开关灯|线段树(带lazy)

题目描述 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

#include<iostream>
#include<cstdio>
using namespace std;

int c,x,y,n,m;
struct tree{
   int l,r,v,lazy;
}t[800001];

void build(int now,int x,int y)
{
    t[now].l=x; t[now].r=y;
    if (x==y) return;
    int mid=(x+y)>>1;
    build(now<<1,x,mid);
    build(now<<1|1,mid+1,y);
}

void pushdown(int k)
{
    if (!t[k].lazy) return;
    t[k<<1].lazy=!t[k<<1].lazy;
    t[k<<1|1].lazy=!t[k<<1|1].lazy;
    t[k<<1].v=t[k<<1].r-t[k<<1].l+1-t[k<<1].v;
    t[k<<1|1].v=t[k<<1|1].r-t[k<<1|1].l+1-t[k<<1|1].v;
    t[k].lazy=0;
}

void change(int k,int x,int y)
{
    pushdown(k);
    int l=t[k].l,r=t[k].r;
    if (l==x && r==y)
     {
          t[k].v=y-x+1-t[k].v;
          if(l!=r) t[k].lazy=1;
          return;
        }
    int mid=(l+r)>>1;
    if (mid>=y) change(k<<1,x,y);
        else if (mid<x) change(k<<1|1,x,y);
        else 
        {
            change(k<<1,x,mid);
            change(k<<1|1,mid+1,y);
        }
    t[k].v=t[k<<1].v+t[k<<1|1].v;
}

int ask(int k,int x,int y)
{
    pushdown(k);
    int l=t[k].l,r=t[k].r;
    if (l==x && y==r) return t[k].v;
    int mid=(l+r)>>1;
    if (mid>=y) return ask(k<<1,x,y);
        else if (mid<x) return ask(k<<1|1,x,y);
            else return (ask(k<<1,x,mid)+ask(k<<1|1,mid+1,y));
}

int main()
{
    cin >> n >> m;
    build(1,1,n);
    for (int i=1; i<=m; i++)
    {
        cin >> c >> x >> y;
        if (c==0) change(1,x,y);
            else cout << ask(1,x,y);
    }
    return 0;
}
原文地址:https://www.cnblogs.com/Shymuel/p/4645564.html