【Vijos-P1512】SuperBrother打鼹鼠-二维树状数组
题目大意:大致与HOJ1640相同。
做法:基本的二维树状数组操作,我在HOJ1640已经讲过了,这里就不多讲了。
以下是本人代码:
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int s,op;
long c[1050][1050]={0};
int lowbit(int i)
{
return i&(-i);
}
void add(int x,int y,int a)
{
for(int i=x;i<=s;i+=lowbit(i))
for(int j=y;j<=s;j+=lowbit(j))
c[i][j]+=a;
}
long sum(int x,int y)
{
long sm=0;
if (!x||!y) return 0;
for(int i=x;i>0;i-=lowbit(i))
for(int j=y;j>0;j-=lowbit(j))
sm+=c[i][j];
return sm;
}
long query(int l,int b,int r,int t)
{
return sum(r,t)-sum(r,b-1)-sum(l-1,t)+sum(l-1,b-1);
}
int main()
{
scanf("%d",&s);
while(scanf("%d",&op)!=EOF)
{
if (op==1)
{
int x,y,a;
scanf("%d %d %d",&x,&y,&a);
add(x+1,y+1,a);
}
if (op==2)
{
int l,b,r,t;
scanf("%d %d %d %d",&l,&b,&r,&t);
printf("%ld\n",query(l+1,b+1,r+1,t+1));
}
}
return 0;
}