poj1195 Mobile phones 树状数组

#include <iostream>
#include <cstring>
#include <cmath>
#define N 1028
#define lowbit(x) x&(-x)
using namespace std;
int num,n,q,x1,x2,y3,y2;
int c[N][N];
void add(int x,int y,int num)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        for(int k=y;k<=n;k+=lowbit(k))
        {
            c[i][k]+=num;
        }
    }
}
int query(int x,int y)
{
    int sum=0;
     for(int i=x;i>0;i-=lowbit(i))
    {
        for(int k=y;k>0;k-=lowbit(k))
        {
           sum+=c[i][k];
        }
    }
    return sum;
}
int main(){
    int num;
    while(cin>>num)
    {
        if(num==3) break;

        else if(num==0)
        {
            cin>>n;
            memset(c,0,sizeof(c));
        }
        else if(num==1)
        {
            cin>>x1>>y3>>q;

           // add(x1,y3,q); 错误
               add(x1+1,y3+1,q);因为在add函数中i k 都要>0 不能>=0 否则会出错 所以要将整个二维挪一下位置  }
        else if(num==2) { 
         cin>>x1>>y3>>x2>>y2; //cout<<abs(query(x2,y2)+query(x1-1,y3-1)-query(x1-1,y2-1)-query(x2-1,y3-1))<<endl;错误
         cout<<abs(query(x2,y2)+query(x1-1,y3-1)-query(x1,y2-1)-query(x2-1,y3))<<endl;
} } return 0;}

posted @ 2018-03-22 21:02  LandingGuys  阅读(91)  评论(0编辑  收藏  举报