poj1656----二维线段树

二维线段树有树套树和四分法来写，表示小菜只会用树套树，而且在一维的时候lazy标记还是不太会写，，囧大了，，，

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 105
struct SNode
{
	int l,r;
	bool flag,color;
};
struct TNode
{
	int l,r;
	SNode son[4*maxn];
}nod[4*maxn];
int n,m;
void subuild(int fa,int root,int l,int r)
{
	SNode *tmp=&nod[fa].son[root];
	tmp->l=l;tmp->r=r;tmp->flag=1;tmp->color=0;
	if(l==r) return ;
	int mid=(l+r)>>1,p=root<<1;
	subuild(fa,p,l,mid);
	subuild(fa,p+1,mid+1,r);
}
void build(int root,int l,int r)
{
	subuild(root,1,1,100);
	nod[root].l=l;nod[root].r=r;
	if(l==r) return ;
	int p=root<<1,mid=(l+r)>>1;
	build(p,l,mid);
	build(p+1,mid+1,r);
}
void subinsert(int fa,int root,int l,int r,bool val)
{
	SNode *tmp=&nod[fa].son[root];
	if(tmp->l==l&&tmp->r==r)
	{
		tmp->flag=1;tmp->color=val;
		return ;
	}
	int mid=(tmp->l+tmp->r)>>1,p=root<<1;
	if(tmp->flag)
	{
		SNode *tmp1=&nod[fa].son[p];
		SNode *tmp2=&nod[fa].son[p+1];
		tmp1->flag=tmp2->flag=1;
		tmp1->color=tmp2->color=tmp->color;
		tmp->flag=0;
	}
	if(r<=mid) subinsert(fa,p,l,r,val);
	else if(l>mid) subinsert(fa,p+1,l,r,val);
	else
	{
		subinsert(fa,p,l,mid,val);
		subinsert(fa,p+1,mid+1,r,val);
	}
}
void insert(int root,int l,int r,int x,int y,bool val)
{
	if(nod[root].l==l&&nod[root].r==r&&l==r)
		subinsert(root,1,x,y,val);
	else
	{
		int mid=(nod[root].l+nod[root].r)>>1,p=root<<1;
		if(r<=mid) insert(p,l,r,x,y,val);
		else if(l>mid) insert(p+1,l,r,x,y,val);
		else
		{
			insert(p,l,mid,x,y,val);
			insert(p+1,mid+1,r,x,y,val);
		}
	}
}
int subquery(int fa,int root,int l,int r)
{
	SNode *tmp=&nod[fa].son[root];
	if(tmp->l==l&&tmp->r==r&&tmp->flag)
	{
		if(tmp->color) return tmp->r-tmp->l+1;
		else return 0;
	}
	int mid=(tmp->l+tmp->r)>>1,p=root<<1;
	if(tmp->flag)
	{
		SNode *tmp1=&nod[fa].son[p];
		SNode *tmp2=&nod[fa].son[p+1];
		tmp1->flag=tmp2->flag=1;
		tmp1->color=tmp2->color=tmp->color;
		tmp->flag=0;
	}
	if(r<=mid) return subquery(fa,p,l,r);
	else if(l>mid) return subquery(fa,p+1,l,r);
	else
	{
		return subquery(fa,p,l,mid)+subquery(fa,p+1,mid+1,r);
	}
}
int query(int root,int l,int r,int x,int y)
{
	if(nod[root].l==l&&nod[root].r==r&&l==r)
		return subquery(root,1,x,y);
	int mid=(nod[root].l+nod[root].r)>>1,p=root<<1;
	if(r<=mid) return query(p,l,r,x,y);
	else if(l>mid) return query(p+1,l,r,x,y);
	else return query(p,l,mid,x,y)+query(p+1,mid+1,r,x,y);
}
int main()
{
	int i,j,k,x,y,l;
	char str[20];
	scanf("%d",&n);
	build(1,1,100);
	while(n--)
	{
		scanf("%s%d%d%d",str,&x,&y,&l);
		switch(str[0])
		{
		case 'W':
			insert(1,x,x+l-1,y,y+l-1,0);
			break;
		case 'B':
			insert(1,x,x+l-1,y,y+l-1,1);
			break;
		default:
			printf("%d\n",query(1,x,x+l-1,y,y+l-1));
		}
	}
	return 0;
}