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;
}



发布了55 篇原创文章 · 获赞 45 · 访问量 9万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章