Codeforces 343D Water Tree 题解&代码

题意：首先给出一棵树，这棵树的结点有两种状态，要么空要么有水，然后给出若干的操作

1 x  将x以及x的子孙灌水

2 x 将x以及x的祖先清空

3 x 查询x结点的状态

思路：

其实是树剖QwQ不过个人感觉和线段树还是很像的…所以树剖的本质大约就是dfs序+线段树？

in[]记录节点的节点的dfs编号
out[]记录节点的控制范围
（in[i]<=in[j]<=out[i]当且仅当j节点属于以i为根节点的子树）
pos[i]记录i节点的父亲节点
这样就能用[1，n]表示一颗以1为根节点的树了

1、每次要给i节点加水的时候就把[in[i],out[i]]的值修改为1，用lazy标记的方式很容易做到
**如果给i节点加水的时候发现有子节点是空的，那么说明i节点的父亲节点一定是空的【否则作为i节点的父亲节点，如果它有水它的子节点一定都是有水的】，如此递推，可以很容易发现此时从i节点父亲向上的祖先全部都是空的，这样就可以把加水后的问题转化为将i的父亲节点清空，父亲节点的查找在dfs时已经提前记录，新问题也就很容易转化为问题2

2、由根向下找到将要清空的点i（即区间[in[i],in[i]]），每一步都pushdown一次，将该点的值修改为0，然后在回溯时每一步都重置节点【如果子节点有空节点，父节点必然是空节点】

3、查询点i状态（区间[in[i],in[i]]），记得每一步都pushdown即可

#include<iostream>
#include<cmath>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<stdlib.h>
#include<string>
#include<algorithm>
#include<queue>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int  maxn =500005;
int sum[maxn<<2];
int add[maxn<<2];
int in[maxn],out[maxn],p[maxn],tot;
int head[maxn];
struct node
{
	int to,next;
}edge[maxn<<2];
int cnt;
int ans=0;
void dfs(int x,int pre)
{
	in[x]=++tot;
	for(int i=head[x];i!=-1;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v!=pre)
		{
			p[v]=x;
			dfs(v,x);
		}
	}
	out[x]=tot;//dfs序列 in[x]~out[x]是x以及x的所有子树
}
void addhead(int u,int v)
{
	edge[cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt;
	cnt++;
	edge[cnt].to=u;
	edge[cnt].next=head[v];
	head[v]=cnt;
	cnt++;
}
void pushup(int rt,int l,int r )//将当前结点的信息更新的父节点
{
	if(l!=r)
	sum[rt]=sum[rt<<1]&&sum[rt<<1|1];//如果子节点都为1  则他为1 否则为0
}
void pushdown(int rt,int l,int r)//延迟标记向下推 m为长度
{
	if(add[rt])
	{
		if(l==r)//叶节点
			sum[rt]=1;
		else
			sum[rt]=sum[rt<<1]=sum[rt<<1|1]=add[rt<<1]=add[rt<<1|1]=1;
	}
	add[rt]=0;
}
void fillwater(int L,int R,int l,int r,int rt)
{
	pushdown(rt,l,r);
	if(L<=l&&r<=R)
	{
		ans=ans&sum[rt];//判断下面是否有0
		sum[rt]=1;
		add[rt]=1;
		pushdown(rt,l,r);
		return ;
	}
	if(r<L||l>R)return ;
	int m=(l+r)>>1;
	fillwater(L,R,lson);
	fillwater(L,R,rson);
	pushup(rt,l,r);
}

void emptywater(int L,int R,int l,int r,int rt)
{
	pushdown(rt,l,r);
	if(r<L||l>R)return ;
	if(l==r)
	{
		//cout<<l<<"***"<<r<<" "<<L<<R<<endl;
		sum[rt]=0;
		return ;
	}

	int m=(l+r)>>1;
	emptywater(L,R,lson);
	emptywater(L,R,rson);
	pushup(rt,l,r);
}
void query(int L,int R,int l,int r,int rt)
{
	pushdown(rt,l,r);
	if(L<=l&&r<=R)
	{
		pushdown(rt,l,r);
		ans=ans&sum[rt];//判断下面是否有0
		return ;
	}
	if(r<L||l>R)return ;
	int m=(l+r)>>1;
	query(L,R,lson);
	query(L,R,rson);

}
int main()
{
	int n;
	int u,v;
	int m;
	int q,c;
	while(~scanf("%d",&n))
	{
		memset(head,-1,sizeof head);
		cnt=0;
		for(int i=0;i<n-1;i++)
		{
			scanf("%d%d",&u,&v);
			addhead(u,v);
		}
		tot=0;
		dfs(1,0);
		/*for(int i=1;i<=n;i++)
		{
			printf("%d %d\n",in[i],out[i]);
		}*/
		scanf("%d",&m);
		for(int i=0;i<m;i++)
		{
			scanf("%d%d",&q,&c);

			if(q==1)//加水
			{
				ans=1;
				fillwater(in[c],out[c],1,tot,1);
				if(!ans&&p[c])
				{
					emptywater(in[p[c]],in[p[c]],1,tot,1);
				}
			}
			else if(q==2)
			{

					emptywater(in[c],in[c],1,tot,1);
			}
			else
			{
				ans=1;
				query(in[c],out[c],1,tot,1);
				printf("%d\n",ans);
			}
			/*for(int j=1;j<=tot;j++)
			{
				ans=1;query(in[j],out[j],1,tot,1) ;
				printf("%d ",ans );
			}
			printf("\n");*/
		}
	}
	return 0;
}