二维树状数组复习—— SuperBrother打鼹鼠

Posted dancer16

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了二维树状数组复习—— SuperBrother打鼹鼠相关的知识,希望对你有一定的参考价值。

       在这个“打鼹鼠”的游戏中,鼹鼠会不时地从洞中钻出来,不过不会从洞口钻进去(鼹鼠真胆大……)。洞口都在一个大小为n(n< =1024)的正方形中。这个正方形在一个平面直角坐标系中,左下角为(0,0),右上角为(n-1,n-1)。洞口所在的位置都是整点,就是横纵坐标都为整数的点。而SuperBrother也不时地会想知道某一个范围的鼹鼠总数。这就是你的任务。

输入

每个输入文件有多行。 第一行,一个数n,表示鼹鼠的范围。 以后每一行开头都有一个数m,表示不同的操作: m=1,那么后面跟着3个数x,y,k(0< =x,y< n),表示在点(x,y)处新出现了k只鼹鼠; m=2,那么后面跟着4个数x1,y1,x2,y2(0< =x1< =x2< n,0< =y1< =y2< n),表示询问矩形(x1,y1)-(x2,y2)内的鼹鼠数量; m=3,表示老师来了,不能玩了。保证这个数会在输入的最后一行。 询问数不会超过10000,鼹鼠数不会超过maxlongint。

输出

对于每个m=2,输出一行数,这行数只有一个数,即所询问的区域内鼹鼠的个数。

样例输入

4
1 2 2 5
2 0 0 2 3
3

样例输出

5

提示

 

水题一道。 所有数据均为随机生成,包括样例……

 题目很好懂,暴力可以水。为复习写了二维树状数组。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define maxn 1025
#define lowbit(x) x&(-x)
int d[4][maxn][maxn],n;
int getsum(int id,int x,int y)
{
    int sum=0;
    for(int i=x;i;i-=lowbit(i))
        for(int j=y;j;j-=lowbit(j))
          sum+=d[id][i][j];
    return sum;      
}
int ask(int x,int y)
{
    return (getsum(0,x,y)*(x+1)*(y+1)-(y+1)*getsum(1,x,y)-(x+1)*getsum(2,x,y)+getsum(3,x,y));
}
void U(int id,int x,int y,int del)
{
    for(int i=x;i<=n;i+=lowbit(i))
        for(int j=y;j<=n;j+=lowbit(j))
            d[id][i][j]+=del;
}
void modify(int x,int y,int c)
{
    U(0,x,y,c);U(0,x+1,y,-c);U(0,x,y+1,-c);U(0,x+1,y+1,c);
    U(1,x,y,x*c);U(1,x+1,y,-(x+1)*c);U(1,x,y+1,-x*c);U(1,x+1,y+1,(x+1)*c);
    U(2,x,y,y*c);U(2,x+1,y,-y*c);U(2,x,y+1,-(y+1)*c);U(2,x+1,y+1,(y+1)*c);
    U(3,x,y,x*y*c);U(3,x+1,y,-(x+1)*y*c);U(3,x,y+1,-x*(y+1)*c);U(3,x+1,y+1,(x+1)*(y+1)*c);
}
int main()
{
    scanf("%d",&n);
    int pattern;
    while(scanf("%d",&pattern)&&pattern!=3)
    {
        if(pattern==1)
        {
            int x,y,c;
            scanf("%d %d %d",&x,&y,&c);
            x++,y++;
            modify(x,y,c);
        }else
        {
            int a,b,c,d;
            scanf("%d %d %d %d",&a,&b,&c,&d);
            a++,b++,c++,d++;
            int ans=ask(c,d)-ask(a-1,d)-ask(c,b-1)+ask(a-1,b-1);
            printf("%d\n",ans); 
        }
    }
}

 

以上是关于二维树状数组复习—— SuperBrother打鼹鼠的主要内容,如果未能解决你的问题,请参考以下文章

[Vijos1512] SuperBrother打鼹鼠 (二维树状数组)

Vijos1512 SuperBrother打鼹鼠

树状数组求区间最大值(树状数组)(复习)

二维树状数组及(不会用到的)三维树状数组

树状数组 / 二维树状数组

树状数组及二维树状数组