Codeforces Round 200 Div1 D Water Tree (树上线段树)

     本来早就应该做的，拖拖拖一直拖到了现在。这个题信息量真大。要注意树上的点虽然有一个区间，但是在线段树中代表这个点的还是一个点，而不是一段区间。

     树上的括号序列，就是按照 dfs 的时序，存在包含关系的节点有祖先儿子关系。

     这个题中有三个操作 1 是向某节点加水， 2 是抽水 ， 3 是观察

     加水之后，所有的儿子也都加满水，放水之后，所有的父亲也都被放水。

     这里 data[x][0]  代表 x 是最后被加满时间 data[x][1] 代表最后被放空时间。

     这里查询的时候，要分清楚是单点的查询还是区间的查询

      查询是否是满的时候因为不和儿子有关，所以，查询到某个点。

     查询是否式空的时候，因为若儿子空了，那么父亲就空了，所有查询一个区间中的最大值。

     虽然原来树中的每个点都有一个区间，但是在线段树中代表这个点的还是一个点，而不是线段树中的非叶子节点。这就相当于把树扯平了。该查点查点，该查区间查区间

     第一个方法是利用两棵线段树，分别记录灌水和抽干的时间，比较这两个时间的前后。

     第二个方法是只使用一棵线段树，记录是否为干。因为干的非常好查询，对于一个点所包含的区间，若其中有一个为干，则这个点便为干，否则这个点就是满的。利用了二进制的性质。另外，从这个题可以看到，线段树合适的题目大多是查询“存在”问题的。而不是“所有”问题。不过注意，当一个点为空的时候将他加满之前要把他的父节点变成空的。

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
using namespace std;
#define MAX 500009
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

vector<int> G[MAX];
int L[MAX];
int R[MAX];
int cnt=0;
int data[MAX<<3][2];// 0 fill 1 empty
void dfs(int x,int fa)
{
	L[x]=++cnt;
	for(int i=0;i<G[x].size();i++)
	{
		int v=G[x][i];
		if(v!=fa)
			dfs(v,x);
	}
	R[x]=++cnt;
}

void PushDown(int rt)
{
		data[rt<<1][0] = max(data[rt<<1][0],data[rt][0]);
		data[rt<<1|1][0] = max(data[rt<<1|1][0],data[rt][0]);
}
void PushUp(int rt)
{
	int t=max(data[rt<<1][1],data[rt<<1|1][1]);
	data[rt][1]=max(t,data[rt][1]);
}
void Update(int a,int b,int op,int time,int l,int r,int rt)
{

	if(b<l||a>r) return;
//	printf("a %d b %d op %d time %d l %d r %d rt %d\n",a,b,op,time,l,r,rt);
	if(a<=l&&r<=b)
	{
		data[rt][op]=time;
		return;
	}
	PushDown(rt);
	int m = (l+r)>>1;
	Update(a,b,op,time,lson);
	Update(a,b,op,time,rson);
	PushUp(rt);
}
int Query(int a,int b,int op,int l,int r,int rt)
{

	if(a>r||b<l)
		return -1;
	if(a<=l&&r<=b)
		return 	data[rt][op];
	
	PushDown(rt);
	
	int m = (l+r)>>1;
	int ans = 0;
	ans=max(ans,Query(a,b,op,lson));
	ans=max(ans,Query(a,b,op,rson));
	PushUp(rt);
//	printf("v1 %d v2 %d\n", v1, v2);
	return ans;	
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=0;i<n-1;i++)
	{
		int f,t;
		scanf("%d%d",&f,&t);
		G[f].push_back(t);
		G[t].push_back(f);
	}
	dfs(1,0);
	memset(data,0,sizeof(data));
	int m;
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		int t,v;
		scanf("%d%d",&t,&v);
		switch(t)
		{
			case 1:
				Update(L[v],R[v],0,i,L[1],R[1],1);
				break;
			case 2:
				Update(L[v],L[v],1,i,L[1],R[1],1);
				break;		
			case 3:
				int v1=Query(L[v],L[v],0,L[1],R[1],1);
				int v2=Query(L[v],R[v],1,L[1],R[1],1);
				if(v1>v2)
					puts("1");
				else
					puts("0");
				break;
		}
	}
	return 0;
}

单棵树版本

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
using namespace std;

#define MAX 500003
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int data[(MAX<<3)+MAX];
vector<int> G[MAX];
int L[MAX];
int R[MAX];
int Fill[(MAX<<3)+MAX];
int cnt;
int pre[MAX];
void PushUp(int rt)
{
	data[rt]=max(data[rt<<1],data[rt<<1|1]);
}

void PushDown(int rt)
{
	if(Fill[rt])
	{
		Fill[rt<<1|1]=Fill[rt<<1]=1;
		data[rt<<1|1]=data[rt<<1]=0;
		Fill[rt]=0;
	}
}

// 0 是满 1 是空
void Update(int a,int b,int op,int l,int r,int rt)
{
	if(b<l||a>r)
		return ;
	if(a<=l&&r<=b)
	{
		data[rt]=op;
		if(!op)
			Fill[rt]=1;
		else
			PushDown(rt);
		return;
	}
	PushDown(rt);
	int m=(l+r)>>1;
	Update(a,b,op,lson);
	Update(a,b,op,rson);
	PushUp(rt);
}

void Build(int l,int r,int rt)
{
	if(l==r)
	{
		data[rt]=1;
		return;
	}
	int m=(l+r)>>1;
	Build(lson);
	Build(rson);
	PushUp(rt);
}

int Query(int a,int b,int l,int r,int rt)
{
	if(b<l||a>r)
		return 0;
	if(a<=l&&r<=b)
	{	
		return data[rt];
	}
	PushDown(rt);
	int m=(l+r)>>1;
	int v1=Query(a,b,lson);
	int v2=Query(a,b,rson);
	PushUp(rt);	
	return max(v1,v2);
}
void AddEdge(int f,int t)
{
	G[f].push_back(t);
	G[t].push_back(f);
}
void dfs(int x,int fa)
{
	pre[x]=fa;
	L[x]=++cnt;
	for(int i=0;i<G[x].size();i++)
	{
		int v=G[x][i];
		if(v!=fa)
			dfs(v,x);
	}
	R[x]=++cnt;
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=0;i<n-1;i++)
	{
		int f,t;
		scanf("%d%d",&f,&t);
		AddEdge(f,t);
	}
	dfs(1,0);
	Build(L[1],R[1],1);
	int m;
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		int t,v;
		scanf("%d%d",&t,&v);
		switch(t)
		{
			case 1:
				if(Query(L[v],R[v],L[1],R[1],1))
					Update(L[pre[v]],L[pre[v]],1,L[1],R[1],1);		
				Update(L[v],R[v],0,L[1],R[1],1);
				break;
			case 2:
				Update(L[v],L[v],1,L[1],R[1],1);
				break;
			case 3:
				if(Query(L[v],R[v],L[1],R[1],1))
					puts("0");
				else
					puts("1");
				break;
		}
	}
	return 0;
}