Vijos / 题库 /
SuperBrother打鼹鼠
背景
SuperBrother在机房里闲着没事干(再对比一下他的NOIP,真是讽刺啊……),于是便无聊地开始玩“打鼹鼠”……
描述
在这个“打鼹鼠”的游戏中,鼹鼠会不时地从洞中钻出来,不过不会从洞口钻进去(鼹鼠真胆大……)。洞口都在一个大小为n(n<=1024)的正方形中。这个正方形在一个平面直角坐标系中,左下角为(0,0),右上角为(n-1,n-1)。洞口所在的位置都是整点,就是横纵坐标都为整数的点。而SuperBrother也不时地会想知道某一个范围的鼹鼠总数。这就是你的任务。
格式
输入格式
每个输入文件有多行。
第一行,一个数n,表示鼹鼠的范围。
以后每一行开头都有一个数m,表示不同的操作:
m=1,那么后面跟着3个数x,y,k(0<=x,y
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
int n;
int c[1050][1050];
inline int lowbit(int x){
return x&-x;
}
inline void add(int x,int y,int val){
for(int i=x;i<=n;i+=lowbit(i)){
for(int j=y;j<=n;j+=lowbit(j)){
c[i][j]+=val;
}
}
}
inline int ask(int x,int y){//求这个矩阵里的和
int res=0;
for(int i=x;i>0;i-=lowbit(i)){
for(int j=y;j>0;j-=lowbit(j)){
res+=c[i][j];
}
}
return res;
}
int main(){
scanf("%d",&n);
int m;
while(scanf("%d",&m)==1){
if(m==3) return 0;
else if(m==1){
int x,y,k;
scanf("%d%d%d",&x,&y,&k);
add(x+1,y+1,k);
}
else{
int x1,y1,y2,x2;
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
printf("%d\n",ask(x2+1,y2+1)-ask(x2+1,y1)-ask(x1,y2+1)+ask(x1,y1));
}
}
}
二维树状数组不能从0开始存~~~~