poj 1195 - 二维线段树

 

解题思路：

二维线段树入门题点击入门

 

//poj 1195
#include<cstdio>
#include<iostream>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
using namespace std;
typedef long long ll;
const int mx = 1e3 + 100;
int sum[mx*3][mx*3],m,n;
int L1,R2,L2,R1,u,flag;
void update_y(int l,int r,int rt,int d)
{
	if(l==r&&flag){ sum[d][rt] += u; return ; }
	if(l==r){
		sum[d][rt] = sum[d<<1][rt] + sum[d<<1|1][rt];
		return ;
	}
	int mid = (l+r)>>1;
	if(R1<=mid) update_y(lson,d);
	else update_y(rson,d);
	sum[d][rt] = sum[d][rt<<1] + sum[d][rt<<1|1];
}
void update_x(int l,int r,int rt)
{
	if(l==r){
		update_y(1,m,1,rt);	flag = 0;
		return ;
	}
	int mid = (l+r)>>1;
	if(L1<=mid) update_x(lson);
	else update_x(rson);
	update_y(1,m,1,rt);
}
int query_y(int l,int r,int rt,int d)
{
	if(L2<=l&&R2>=r) return sum[d][rt];
	int mid = (l+r)>>1,ans = 0;
	if(L2<=mid) ans += query_y(lson,d);
	if(R2>mid) ans += query_y(rson,d);
	return ans;
}
int query_x(int l,int r,int rt)
{
	if(L1<=l&&R1>=r) return query_y(1,m,1,rt);
	int mid = (l+r)>>1,ans = 0;
	if(L1<=mid) ans += query_x(lson);
	if(R1>mid) ans += query_x(rson);
	return ans;
}
int main()
{
	while(scanf("%d",&n)){
		flag = 1;
		if(n==3) break;
		if(!n)	scanf("%d",&m);
		else if(n==1){
			scanf("%d%d%d",&L1,&R1,&u);
			L1++,R1++;
			update_x(1,m,1);
		}else{
			scanf("%d%d%d%d",&L1,&L2,&R1,&R2);
			L1++,L2++,R1++,R2++;
			printf("%d\n",query_x(1,m,1));
		}
	}
	return 0;
}