POJ 1195 mobile phones 二维树状数组

题意很简单，就是求一定范围内子矩阵的和，数据量很大，用暴力的话一定会超时，这道题用二维的树状数组做比较简单。。。

#include<stdio.h>

#include<string.h>

#include <iostream>

using namespace std;

#define MAX 1030

int tree[MAX][MAX];

int S,ans;

int lowbit(int x){return x&(-x);}

void add(int x,int y,int num)

{

	for(int i=x;i<=S;i+=lowbit(i))

	{

		for(int j=y;j<=S;j+=lowbit(j))

		{

			tree[i][j]+=num;

		}

	}

}

int query(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+=tree[i][j];

		}

	}

	return sum;

}

int main()

{

	setbuf(stdout,NULL);

	int instruction;

	scanf("%d %d",&instruction,&S);

	memset(tree,0,sizeof(tree));



	while(scanf("%d",&instruction)!=EOF)

	{

		if(instruction==3)break;

		if(instruction==1)

		{

			int X,Y,A;

			scanf("%d %d %d",&X,&Y,&A);

			add(X+1,Y+1,A);

		}

		else

		{

			int L,B,R,T;

			scanf("%d %d %d %d",&L,&B,&R,&T);

			ans=query(R+1,T+1)+query(L,B)-query(L,T+1)-query(R+1,B);

			printf("%d\n",ans);

		}

	}

	return 0;

}