二維樹狀數組(poj 1195)

題目:http://poj.org/problem?id=1195

題意:

給出一個全0的矩陣,然後一些操作
0 S:初始化矩陣,維數是S*S,值全爲0,這個操作只有最開始出現一次
1 X Y A:對於矩陣的X,Y座標增加A
2 L B R T:詢問(L,B)到(R,T)區間內值的總和
3:結束對這個矩陣的操作

題解:利用二維樹狀數組,利用logn複雜度對一個點進行更新,再利用logn複雜度踐行查找,題目就很簡單了

代碼:

#include<cstdio>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>

using namespace std;
int c[1500][1500];
int MAXN=1025;
//樹狀數組常用的東西,不加多說
int lowbit(int x)
{
    return x & (-x);
}
//注意 該更新是二維更新,複雜度logn × logn
int add(int x,int y,int val)
{

    for(int i=x;i<=MAXN;i+=lowbit(i))
    {
        for(int j=y;j<=MAXN;j+=lowbit(j))
        {
            c[i][j]+=val;
        }
    }
}
//求和操作
int sum(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 l,r;
    scanf("%d %d",&l,&r);
    MAXN=r;
    int q;
    while(~scanf("%d",&q) && q!=3)
    {
        if(q==1)
        {
            int x,y,val;
            scanf("%d %d %d",&x,&y,&val);
            x++;y++;
            add(x,y,val);
        }
        else if(q==2)
        {
            int l1,l2,r1,r2;
            scanf("%d%d%d%d",&x,&y,&l1,&r1);
            x++;
            y++;
            l1++;
            r1++;
            printf("%d\n",sum(l1,r1)-sum(l1,y-1)-sum(x-1,r1)+sum(x-1,y-1));
        }
    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章