Vijos1512SuperBrother打鼹鼠

背景
SuperBrother在机房里闲着没事干(再对比一下他的NOIP,真是讽刺啊……),于是便无聊地开始玩“打鼹鼠”……
描述
在这个“打鼹鼠”的游戏中,鼹鼠会不时地从洞中钻出来,不过不会从洞口钻进去(鼹鼠真胆大……)。洞口都在一个大小为n(n<=1024)的正方形中。这个正方形在一个平面直角坐标系中,左下角为(0,0),右上角为(n-1,n-1)。洞口所在的位置都是整点,就是横纵坐标都为整数的点。而SuperBrother也不时地会想知道某一个范围的鼹鼠总数。这就是你的任务。
格式
输入格式

每个输入文件有多行。
第一行,一个数n,表示鼹鼠的范围。
以后每一行开头都有一个数m,表示不同的操作:
m=1,那么后面跟着3个数x,y,k(0<=x,y

#include
using namespace std;
int n,f[1030][1030]={0},m;

int read()
{
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0' && ch>'9')
      {
        if (ch=='-')
          f=-1;
        ch=getchar();
      }
    while (ch>='0' && ch<='9')
      {
        x=x*10+ch-'0';
        ch=getchar();
      }
    return x*f;
}

int lowbit(int x)
{
    return x&(-x);
}

void add(int x,int y,int t)
{
    int i,j;
    for (i=x;i<=n;i+=lowbit(i))
      for (j=y;j<=n;j+=lowbit(j))
        f[i][j]+=t;
}

int query(int x,int y)
{
    int i,j,ans=0;
    for (i=x;i>=1;i-=lowbit(i))
      for (j=y;j>=1;j-=lowbit(j))
        ans+=f[i][j];
    return ans;
}

int main()
{
    int x,y,x1,y1,t;
    n=read();
    while (m=read(),m!=3)
      {
        if (m==1)
          {
            x=read()+1;
            y=read()+1;
            t=read();
            add(x,y,t);
          }
        if (m==2)
          {
            x=read()+1;
            y=read()+1;
            x1=read()+1;
            y1=read()+1;
            printf("%d\n",query(x1,y1)-query(x-1,y1)-query(x1,y-1)+query(x-1,y-1));
          }
      }
    return 0;
}

你可能感兴趣的:(树状数组)