洛谷P2826 [USACO08NOV]光开关Light Switching [2017年6月计划 线段树02]

P2826 [USACO08NOV]光开关Light Switching

题目描述

Farmer John tries to keep the cows sharp by letting them play with intellectual toys. One of the larger toys is the lights in the barn. Each of the N (2 <= N <= 100,000) cow stalls conveniently numbered 1..N has a colorful light above it.

At the beginning of the evening, all the lights are off. The cows control the lights with a set of N pushbutton switches that toggle the lights; pushing switch i changes the state of light i from off to on or from on to off.

The cows read and execute a list of M (1 <= M <= 100,000) operations expressed as one of two integers (0 <= operation <= 1).

The first kind of operation (denoted by a 0 command) includes two subsequent integers S_i and E_i (1 <= S_i <= E_i <= N) that indicate a starting switch and ending switch. They execute the operation by pushing each pushbutton from S_i through E_i inclusive exactly once.

The second kind of operation (denoted by a 1 command) asks the cows to count how many lights are on in the range given by two integers S_i and E_i (1 <= S_i <= E_i <= N) which specify the inclusive range in which the cows should count the number of lights that are on.

Help FJ ensure the cows are getting the correct answer by processing the list and producing the proper counts.

灯是由高科技——外星人鼠标操控的。你只要左击两个灯所连的鼠标，

这两个灯，以及之间的灯都会由暗变亮，或由亮变暗。右击两个灯所连的鼠

标，你就可以知道这两个灯，以及之间的灯有多少灯是亮的。你的任务是在LZ

之前算出灯的亮灭。

输入输出格式

输入格式：

第1 行: 用空格隔开的两个整数N 和M，n 是灯数

第2..M+1 行: 每行表示一个操作, 有三个用空格分开的整数: 指令号, S_i 和E_i

第1 种指令(用0 表示)包含两个数字S_i 和E_i (1 <= S_i <= E_i <= N), 它们表示起

始开关和终止开关. 表示左击

第2 种指令(用1 表示)同样包含两个数字S_i 和E_i (1 <= S_i <= E_i <= N), 不过这

种指令是询问从S_i 到E_i 之间的灯有多少是亮着的.

输出格式：

输入输出样例

输入样例#1：

4 5
0 1 2
0 2 4
1 2 3
0 2 4
1 1 4

输出样例#1：

1
2

说明

原题时间限制为2s，内存限制为16M

至今才知道原来修改操作也需要下放标记，这样省去之前很多很麻烦的判断写法了T.T

利用异或的特性减少了很多if判断和修改。用1和0不断异或即可。

线段树记录区间1的数量，lazy标记记录当前字数是否应该反转

0^1 = 1 1^1 = 0 。。。。

 #include <bits/stdc++.h>
 inline void read(int &x)
 {
     x = ;char ch = getchar();char c = ch;
     while(ch > '' || ch < '')c = ch, ch = getchar();
     while(ch <= '' && ch >= '')x = x *  + ch - '',ch = getchar();
     if(c == '-')x = -x;
 }
 const int MAXN =  + ;
 const int MAXM =  + ;
 inline void swap(int &a,int &b)
 {
     int tmp = a;
     a = b;
     b = tmp;
 }

 int stdata[MAXN << ];int lazy[MAXN << ];
 //线段树存数字1的个数，lazy:0则不动， 1则反置 

 int n,m;

 void putdown(int o, int l, int r)
 {
     int mid = (l + r) >> ;
     lazy[o << ] ^= lazy[o];
     lazy[o <<  | ] ^= lazy[o];
     lazy[o] = ;
     stdata[o << ] = (mid - l + ) - stdata[o << ];
     stdata[o <<  | ] = (r - mid) - stdata[o <<  | ];
 } 

 void modify(int ll, int rr, int o = , int l = , int r = n)
 {
     if(ll <= l && rr >= r)
     {
         stdata[o] = (r - l + ) - stdata[o];
         lazy[o] ^= ;
         return;
     }
     int mid = (l + r) >> ;
     if(lazy[o])putdown(o, l, r);
     if(mid >= ll)modify(ll, rr, o << , l, mid);
     if(mid < rr) modify(ll, rr, o <<  | , mid + , r);
     stdata[o] = stdata[o << ] + stdata[o <<  | ];
 }

 int query(int ll, int rr, int o = , int l = , int r = n)
 {
     if(ll <= l && rr >= r)
     {
         return stdata[o];
     }
     int ans = ;
     int mid = (l + r) >> ;
     if(lazy[o]) putdown(o, l, r);
     if(mid >= ll) ans += query(ll, rr, o << , l, mid);
     if(mid < rr)ans += query(ll, rr, o <<  | , mid + , r);
     return ans;
 }

 int main()
 {
     read(n);read(m);
     for(int i = ;i <= m;i ++)
     {
         int tmp1,tmp2,tmp3;
         read(tmp1);read(tmp2);read(tmp3);
         if(tmp1 == )
         {
             modify(tmp2, tmp3);
         }
         else
         {
             printf("%d\n", query(tmp2, tmp3));
         }
     }
     return ;
 }