POJ 1195 Mobile phones 二维树状数组模板题

题意：

给出一个全0的矩阵，然后一些操作

0 S：初始化矩阵，维数是S*S，值全为0，这个操作只有最开始出现一次

1  x  y  val：对于矩阵的X,Y坐标增加A

2  x1  y1  x2  y2：询问(x1,y1)到(x2,y2)区间内值的总和

3：结束对这个矩阵的操作

 

思路：

使用二维树状数组进行操作

AC_code

#include<iostream>
#include<iomanip>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<queue>
#include<stack>
#include<vector>
#include<map>
#include<cstdlib>
#define ll long long
#define inf 0x3f3f3f3f//1e9+6e7
#define eps 1e-8
#define mod 1e9+7
#define N 1110
#define mem(a,x) memset((a),(x),sizeof ((a)))//x只能是0或-1
#define debug(x) cout<<"X: "<<(x)<<endl
#define de cout<<"************"<<endl
#define lowbit(x) ((x)&(-x))
#define lson rt<<1
#define rson rt<<1|1
#define gcd(a,b) __gcd(a,b)
#define lcm(a,b) a*b/(__gcd(a,b))
const double pi=acos(-1.0);
using namespace std;

int c[N][N];
int n;
void update(int x,int y,int d)
{
    for(int i=x;i<=N;i+=lowbit(i))  
    {
        //循环条件是i小于等于n还是小于等于N，这个我不太确定，但是这个题n或N都能过，有些题貌似是只有N才能过，这点我也不太确定
        for(int j=y;j<=N;j+=lowbit(j))
            c[i][j]+=d;
    }
}
int sum(int x,int y)
{
    int ret=0;
    for(int i=x;i>0;i-=lowbit(i))
    {
        for(int j=y;j>0;j-=lowbit(j))
        {
            ret+=c[i][j];
        }
    }
    return ret;
}

int main()
{
    int m;
    int x,y,val;
    int x1,x2,y1,y2;
    while(scanf("%d%d",&m,&n)!=EOF)
    {
        memset(c,0,sizeof c);
        int k;
        while(1)
        {
            scanf("%d",&k);
            if(k==1)
            {
                scanf("%d%d%d",&x,&y,&val);
                x++;
                y++;    //树状数组不能对0操作
                update(x,y,val);
                continue;
            }
            if(k==2)
            {
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                x1++;y1++;x2++;y2++;    //树状数组不能对0操作
                cout<<sum(x2,y2)-sum(x2,y1-1)-sum(x1-1,y2)+sum(x1-1,y1-1)<<endl;
                continue;
            }
            break;
        }
    }
    return 0;
}