简单学习一下虚树,理解了就感觉不是很难。构建的过程很巧妙。
- 将所有点按照 d f s dfs dfs序排序。
- 用一个栈保存一条最右链。在构建过程中,会一边调整栈,一边连边。
-
d
f
s
dfs
dfs序从小到大开始扫描各个关键点。假设当前关键点为
u
u
u。
I . I. I.若栈空或只有一个元素,就把 u u u入栈。
I I . II. II.若栈非空。令 g = l c a ( u , s t k [ t o p ] ) g=lca(u,stk[top]) g=lca(u,stk[top])。
若 g = s t k [ t o p ] g=stk[top] g=stk[top],说明 u u u在 s t k [ t o p ] stk[top] stk[top]的子树中。直接把 u u u入栈。
若 g ≠ s t k [ t o p ] g\not=stk[top] g=stk[top],说明 u u u, s t k [ t o p ] stk[top] stk[top]在 g g g的两个不同子树。那么现在就把在 g g g的子树中的,在栈中的元素弹掉,同时给他们连边。
最后的时候,如果 g ! = t o p g!=top g!=top,那么 g g g和 t o p top top要连上边,并且 g g g要入栈。
然后把 u u u入栈,维护这个最右链。
while(top>1&&dfn[g]<=dfn[stk[top-1]]) re_add(stk[top-1],stk[top]),top--;
if(g!=stk[top]) re_add(g,stk[top]),stk[top]=g;
stk[++top]=u;
虚树插入一个点调用 i n s e r t insert insert函数如下。注意一下,点插入完以后栈里还是可能有东西的。所以最后要把栈里的点都弹出来并建边。求 l c a lca lca可以随便选一个方法。
inline void insert(int u){
if(top<=1){stk[++top]=u;return;}
int g=lca(u,stk[top]);
if(g==stk[top]){stk[++top]=u;return;}
while(top>1&&dfn[g]<=dfn[stk[top-1]])
re_add(stk[top-1],stk[top]),top--;
if(g!=stk[top]) re_add(g,stk[top]),stk[top]=g;
stk[++top]=u;
}
while(--top) re_add(stk[top],stk[top+1]);
警告:注意虚树建图使用的变量名,不要和其他变量搞混了。 i n s e r t insert insert函数可能会因题目中关键点性质的不同而有所变化。
预处理倍增数组、深度数组以及 d f s dfs dfs序。
inline void dfs(int u,int fa){
dfn[u]=++tot,f[u][0]=fa,dep[u]=dep[fa]+1;
for(int i=1;i<=18;++i) f[u][i]=f[f[u][i-1]][i-1];
for(int i=Head[u];i;i=Next[i]) if(V[i]!=fa) dfs(V[i],u);
}
初始定义的变量:
int Head[maxn],Next[maxn<<1],V[maxn<<1],cnt=0;//原图
int head[maxn],nxt[maxn<<1],to[maxn<<1],cntt=0;//虚树
int f[maxn][19],dep[maxn],dfn[maxn],tot=0;
int stk[maxn],x[maxn],top=0;