poj1195 Mobile phones(二維樹狀數組單點更新)

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

思路比較簡單,但是有些細節需要考慮,例如x,y個代表什麼,代碼中代表樹狀數組的第x行第y列,但是題目中的代表的是座標,剛好相反;

座標從0開始,樹狀數組從1開始,組就要用樹狀數組的1代表座標上的0,需要對座標加1;

然後是矩形區域的區域求和;

#include<cstdio>
#include<cstring>
using namespace std;
const int num=1200;
int c[num][num],n,low[num];
void init()
{
    int i;
    for(i=1;i<num;i++)
        low[i]=i&(-i);
}
void add(int x,int y,int a)
{
    int ty;
    while(x<=n)
    {
        ty=y;
        while(ty<=n)
        {
            c[x][ty]+=a;
            ty+=low[ty];
        }
        x+=low[x];
    }
}
int sum(int x,int y)
{
    int s=0,ty;
    while(x>0)
    {
        ty=y;
        while(ty>0)
        {
            s+=c[x][ty];
            ty-=low[ty];
        }
        x-=low[x];
    }
    return s;
}
int main()
{
    int a,b,e,d,p,ans;
    init();
    //freopen("in.txt","r",stdin);
    while(1)
    {
        scanf("%d",&p);
        if(p==0)
        {
            scanf("%d",&n);
            memset(c,0,sizeof(c));
        }
        else if(p==3)
            break;
        else if(p==1)
        {
            scanf("%d%d%d",&a,&b,&e);
            add(b+1,a+1,e);
        }
        else if(p==2)
        {
            scanf("%d%d%d%d",&a,&b,&e,&d);
            a++;b++;e++;d++; //樹狀數組的1表示座標0
            ans=sum(d,e)-sum(d,a-1)-sum(b-1,e)+sum(b-1,a-1);
            printf("%d\n",ans);
        }
    }
    return 0;
}


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