【CodeForces】343D Water Tree

来源：互联网发布：兰州ps软件班编辑：程序博客网时间：2024/05/15 03:47

应该不需要传送门了……

Solution

两个操作，灌水和抽水。
灌水相当于子树覆盖，抽水是链覆盖。
那么维护一下一颗子树中是否有抽过水，在灌水的时候，如果子树中有抽水标记，那么清除标记，并在子树根的父亲的位置打个标记。
【即先抽再灌等于在子树根的父亲的位置抽】

DFN序再用线段树搞搞就好

#include<stdio.h>#define N 500005#define cint const int &int s[N],tot,cnt,sum,dfn[N],ed[N],L,R,aim,n,q,f[N],rt;bool flag;struct edge{int v,n;}e[N<<1];inline void push(cint u,cint v){e[++tot]=(edge){v,s[u]};s[u]=tot;}void dfs(cint k){    ed[k]=dfn[k]=++cnt;    for (int i=s[k];i;i=e[i].n) if (!dfn[e[i].v]) dfs(e[i].v),f[e[i].v]=k,ed[k]=ed[e[i].v];}struct seg{int k,lson,rson;}t[N<<1];void build(int &k,cint l,cint r){    t[k=++sum].k=1;     if (l==r) return;    int mid=l+r>>1;    build(t[k].lson,l,mid);    build(t[k].rson,mid+1,r);}void que(cint k,cint l,cint r){    if (L<=l && r<=R){if (t[k].k) flag=1;return;}    int mid=l+r>>1;    if (L<=mid) que(t[k].lson,l,mid);    if (mid<R) que(t[k].rson,mid+1,r);}void ch1(cint k,cint l,cint r){    if (L<=l && r<=R){if (!t[k].k) return;flag=1;}    if (l==r){t[k].k=0;return;}    int mid=l+r>>1;    if (L<=mid) ch1(t[k].lson,l,mid);    if (mid<R) ch1(t[k].rson,mid+1,r);    t[k].k=t[t[k].lson].k||t[t[k].rson].k;}void ch0(cint k,cint l,cint r){    t[k].k=1;    if (l==r) return;    int mid=l+r>>1;    if (aim<=mid) ch0(t[k].lson,l,mid);    else ch0(t[k].rson,mid+1,r);}int main(){    scanf("%d",&n);    for (int i=1,u,v;i<n;i++) scanf("%d%d",&u,&v),push(u,v),push(v,u);    dfs(1);    build(rt,1,n);    scanf("%d",&q);    for (int type;q--;)    {        scanf("%d%d",&type,&aim);        if (type==1)        {            L=dfn[aim];R=ed[aim];flag=0;ch1(1,1,n);            if (flag && (aim=dfn[f[aim]])) ch0(1,1,n);        }        if (type==2) aim=dfn[aim],ch0(1,1,n);        if (type==3)        {            L=dfn[aim];R=ed[aim];flag=0;            que(1,1,n);            if (flag) puts("0");            else puts("1");        }    }}

0 0