【bzoj 1230】 lites 開關燈 【Usaco2008 Nov】
Description
Farmer John嘗試通過和奶牛們玩益智玩具來保持他的奶牛們思維敏捷. 其中一個大型玩具是牛欄中的燈. N (2 <= N <= 100,000) 頭奶牛中的每一頭被連續的編號為1..N, 站在一個彩色的燈下面.剛到傍晚的時候, 所有的燈都是關閉的. 奶牛們通過N個按鈕來控制燈的開關; 按第i個按鈕可以改變第i個燈的狀態.奶牛們執行M (1 <= M <= 100,000)條指令, 每個指令都是兩個整數中的一個(0 <= 指令號 <= 1). 第1種指令(用0表示)包含兩個數字S_i和E_i (1 <= S_i <= E_i <= N), 它們表示起始開關和終止開關. 奶牛們只需要把從S_i到E_i之間的按鈕都按一次, 就可以完成這個指令. 第2種指令(用1表示)同樣包含兩個數字S_i和E_i (1 <= S_i <= E_i <= N), 不過這種指令是詢問從S_i到E_i之間的燈有多少是亮著的. 幫助FJ確保他的奶牛們可以得到正確的答案.
Input
* 第 1 行: 用空格隔開的兩個整數N和M
* 第 2..M+1 行: 每行表示一個操作, 有三個用空格分開的整數: 指令號, S_i 和 E_i
Output
第 1..詢問的次數 行: 對於每一次詢問, 輸出詢問的結果.
Sample Input
4 5 0 1 2 0 2 4 1 2 3 0 2 4 1 1 4 輸入解釋: 一共有4盞燈; 5個指令. 下面是執行的情況: 燈 1 2 3 4 Init: O O O O O = 關 * = 開 0 1 2 -> * * O O 改變燈 1 和 2 的狀態 0 2 4 -> * O * * 1 2 3 -> 1 輸出在2..3的範圍內有多少燈是亮的 0 2 4 -> * * O O 改變燈 2 ,3 和 4 的狀態 1 1 4 -> 2 輸出在1..4的範圍內有多少燈是亮的
Sample Output
1 2
這道題可以用分塊卡過,對於每個塊,維護當前塊中亮著的燈的數量,每次塊修改就令(為塊大小),下面是程式:
#include<stdio.h> #include<math.h> #include<iostream> #define L(n) (n-1)*m+1 #define R(n) (n)*m #define Q(n) (int)(n*1.0/m+0.99999) using namespace std; const int N=100005; int n,m,q,cnt[355]; bool f[355],a[N]; inline char nc(){ static char buf[100000],*p1=buf,*p2=buf; return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++; } inline void read(int &s){ s=0; char c=nc(); while(c<'0'||c>'9'){ c=nc(); } while(c>='0'&&c<='9'){ s=(s<<3)+(s<<1)+c-'0'; c=nc(); } } inline int min(int &a,int &b){ return a<b?a:b; } void change(int l,int r){ int i,x,y,tp; if(Q(l)==Q(r)){ x=Q(l); y=min(R(x),n); for(i=L(x);i<=y;i++){ a[i]^=f[x]; } f[x]=0; for(i=l;i<=r;i++){ a[i]^=1; } for(i=L(x),tp=0;i<=y;i++){ tp+=a[i]; } cnt[x]=tp; return; } y=Q(r)-1; for(i=Q(l)+1;i<=y;i++){ f[i]^=1; cnt[i]=m-cnt[i]; } x=Q(l); y=R(x); for(i=L(x);i<=y;i++){ a[i]^=f[x]; } f[x]=0; for(i=l;i<=y;i++){ a[i]^=1; } for(i=L(x),tp=0;i<=y;i++){ tp+=a[i]; } cnt[x]=tp; x=Q(r); y=min(n,R(x)); for(i=L(x);i<=y;i++){ a[i]^=f[x]; } f[x]=0; for(i=L(x);i<=r;i++){ a[i]^=1; } for(i=L(x),tp=0;i<=y;i++){ tp+=a[i]; } cnt[x]=tp; } int ask(int l,int r){ int i,x,y,s=0; if((x=Q(l))==Q(r)){ for(i=l;i<=r;i++){ s+=a[i]^f[x]; } return s; } y=Q(r)-1; for(i=Q(l)+1;i<=y;i++){ s+=cnt[i]; } x=Q(l); y=R(x); for(i=l;i<=y;i++){ s+=a[i]^f[x]; } x=Q(r); for(i=L(x);i<=r;i++){ s+=a[i]^f[x]; } return s; } int main(){ int l,r,i; read(n); read(q); m=sqrt(n); while(q--){ read(i); read(l); read(r); if(i){ printf("%d\n",ask(l,r)); } else{ change(l,r); } } return 0; }