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<
c27e
=tot;j++)
{
ans=1;query(in[j],out[j],1,tot,1) ;
printf("%d ",ans );
}
printf("\n");*/
}
}
return 0;
}