【CodeForces】343D Water Tree

应该不需要传送门了……

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");
        }
    }
}