Codeforces 343D Water Tree

题意简述

维护一棵树，支持以下操作：

• 0 v：将以v为跟的子树赋值为1
• 1 v：将v到根节点的路径赋值为0
• 2 v：询问v的值

题解思路

树剖+珂朵莉树

代码

#include <set>
#include <cstdio>
#define IT std::set<Node>::iterator
const int N=500005;
int n,q,u,v,opt,x,cnt;
int h[N],to[N<<1],nxt[N<<1];
int fa[N],sz[N],hvs[N],id[N],top[N];
struct Node {
    int l,r; bool v;
    Node(const int& L,const int& R=-1,const bool& V=0):l(L),r(R),v(V) {}
    bool operator <(const Node& x) const { return l<x.l; }
};
inline void add_edge(const int& u,const int& v) {
    to[++cnt]=v; nxt[cnt]=h[u]; h[u]=cnt;
}
void dfs1(const int& u) {
    sz[u]=1;
    for (register int i=h[u];i;i=nxt[i])
        if (to[i]^fa[u]) {
            fa[to[i]]=u;
            dfs1(to[i]);
            sz[u]+=sz[to[i]];
            if (sz[to[i]]>sz[hvs[u]]) hvs[u]=to[i];
        }
}
void dfs2(const int& u) {
    id[u]=++cnt;
    if (hvs[u]) { top[hvs[u]]=top[u]; dfs2(hvs[u]); }
    for (register int i=h[u];i;i=nxt[i])
        if (to[i]!=hvs[u]&&to[i]!=fa[u]) {
            top[to[i]]=to[i]; dfs2(to[i]);
        }
}
std::set<Node> s;
inline IT split(const int& pos) {
    IT it=s.lower_bound(Node(pos,0,0));
    if (it!=s.end()&&it->l==pos) return it;
    --it;
    const int L=it->l,R=it->r; const bool V=it->v;
    s.erase(it);
    s.insert(Node(L,pos-1,V)); return s.insert(Node(pos,R,V)).first;
}
inline void assign(const int& l,const int& r,const bool& va) {
    IT itr=split(r+1),itl=split(l); s.erase(itl,itr); s.insert(Node(l,r,va));
}
inline void modify(int u) {
    for (;top[u]!=1;u=fa[top[u]]) assign(id[top[u]],id[u],0); assign(1,id[u],0);
}
inline int query(int x) { IT it=split(x); return it->v; }
int main() {
    scanf("%d",&n); s.insert(Node(1,n,0));
    for (register int i=1;i<n;++i)
        scanf("%d%d",&u,&v),add_edge(u,v),add_edge(v,u);
    cnt=0; fa[1]=top[1]=1; dfs1(1); dfs2(1);
    scanf("%d",&q);
    for (register int i=1;i<=q;++i) {
        scanf("%d%d",&opt,&x);
        if (opt==1) assign(id[x],id[x]+sz[x]-1,1);
        else if (opt==2) modify(x);
        else printf("%d\n",query(id[x]));
    }
}