POJ 1195 Mobile phones

題目鏈接戳這裏

題目大意：

初始化一個全部爲0的數組，實現查詢、更新數組的功能；

更新：給出座標(x,y)和一個實數z，把數組中的這個數增加z；

查詢：給出兩組座標(x1,y1),(x2.y2)，統計這個區域所有數的和。

分析：

這道題後臺測試數據量比較大，沒有較好的優化方法很容易超時，如果用簡單的矩陣累加就會超時了；另一種方法就是讓矩陣中每一個定點保存從(0,0)到本身所在位置的這片區域的總和，但這種方法似乎增加了很多不必要的計算；然後二維樹狀數組正好可以用來解決這種問題。套模板即可，需要注意的是，題目給的座標是從0開始的，而樹狀數組中是不能從0開始的。

源代碼：

#include<stdio.h>
#include<cstdio>
#include<string>
#include<cstring>
#include<string.h>
#include<vector>
#include<algorithm>
#include<queue>
#include<stack>
#include<math.h>
#include<cmath>
#include<iomanip>
#include<stdlib.h>
#include<list>
#include<map>
#include<fstream>
#include<ostream>
#include<iostream>
using namespace std;
#define N 1050
int c[N][N],t,s;
int lowbit(int x)
{
	return x&(-x);
}
void add(int x,int y,int xx)
{
	for(int i=x;i<=s;i+=lowbit(i))
		for(int j=y;j<=s;j+=lowbit(j))
			c[i][j]+=xx;
}
int getsum(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 p,x,y,a,l,r;
	scanf("%d%d",&t,&s);
	memset(c,0,sizeof(c));
	while(scanf("%d",&p))
	{
		if(p==3)break;
		else if(p==1)
		{
			scanf("%d%d%d",&x,&y,&a);
			add(x+1,y+1,a);
		}
		else
		{
			scanf("%d%d%d%d",&x,&y,&l,&r);
			x++;y++;l++;r++;
			printf("%d\n",getsum(l,r)-getsum(x-1,r)-getsum(l,y-1)+getsum(x-1,y-1));
		}
	}
	return 0;
}