【2018年全國多校算法寒假訓練營練習比賽(第五場)-E】情人節的電燈泡(二維樹狀數組單點更新+區間查詢)
阿新 • • 發佈：2018-02-27
string 整數 改變 sin microsoft 題目 全國 次數 ros
試題鏈接：https://www.nowcoder.com/acm/contest/77/E
題目描述
情人節到了，小芳和小明手牽手，打算過一個完美的情人節，但是小剛偏偏也來了，當了一個明晃晃的電燈泡，小明很尷尬，就和小剛說，我交給你個任務，你完成了我倆就帶你玩，否則你就回家吧。小剛很有當單身狗的覺悟，他堅決不想讓小明過好情人節，同為單身狗的你能幫幫他嗎？現在有一個n×n（1 <= n <= 1000）的格子，每一個格子都有一個電燈泡，可能是亮的，也可能是滅的（1代表亮， 0代表滅），現在有兩種操作，一種是給你一個坐標，對於那個坐標上的燈泡，如果他是亮的，那麽熄滅他，反之如果他是滅的，那麽打開它。第二種操作是給你兩個坐標，第一個坐標代表一個子矩陣的左上角，另一個坐標是右下角，請你求出當前子矩陣中有多少個燈泡是亮著的。燥起來吧！！！單身狗們！！！！輸入描述:
第一行兩個整數，n（1 <= n <= 1000）和m（1 <= m <= 100000），分別代表正方形格子的邊長和詢問次數。
接下來n行，每一行有n個bool形數字（0或1），代表燈泡的狀態。
接下來m行，每一行第一個數字f（1或2）代表操作的類型，如果f是1，那麽接下來輸入一個坐標（x, y）(1 <= x, y <= n)，對於當前位置的燈泡狀態進行改變，如果是2，那麽接下來輸入兩個坐標（x1, y1）(1 <= x1, y1 <= n)， （x2, y2）(1 <= x2, y2 <= n)，確定子矩陣的位置，輸出子矩陣中亮著的燈泡數量並換行。
輸出描述:
對於每一個2操作，輸出子矩陣中亮著的燈泡數量並換行。
輸入
6 4
0 0 1 0 1 0
1 0 1 1 0 1
1 0 0 0 0 0
1 1 0 0 1 0
0 0 0 0 1 1
0 0 0 0 1 0
2 2 2 4 5
1 1 1
2 1 1 6 6
1 2 6
輸出
4
14
代碼
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #include<cstdlib> #include<cmath> using namespace std; const int N = 1004; typedef long long LL; int c[N][N], n; int lowbit(int t) { return t&(-t); } void update(int x,int y,int m) { for(int i=x; i<=n; i+=lowbit(i)) for(int j=y; j<=n; j+=lowbit(j)) c[i][j]+=m; } int query(int x,int y) { int sum=0; for(int i=x; i>0; i-=lowbit(i)) for(int j=y; j>0; j-=lowbit(j)) sum+=c[i][j]; return sum; } int main() { int i,j,m,k; scanf("%d%d",&n,&m); for(i=1; i<=n; i++) for(j=1; j<=n; j++) { scanf("%d",&k); update(i,j,k); } while(m--) { int t,x,y; scanf("%d", &t); if(t==1) { scanf("%d%d",&x,&y); int s1=query(x,y)-query(x-1,y)-query(x,y-1)+query(x-1,y-1); if(s1) update(x,y,-1); else update(x,y,1); } else { int x1,x2,y1,y2; scanf("%d%d%d%d", &x1,&y1,&x2,&y2); printf("%d\n", query(x2,y2)-query(x1-1,y2)-query(x2,y1-1)+query(x1-1,y1-1)); } } return 0; }
【2018年全國多校算法寒假訓練營練習比賽(第五場)-E】情人節的電燈泡(二維樹狀數組單點更新+區間查詢)