我们用单调栈来建虚树,如果求 lca 的复杂度为 $O(\log n)$ 的话,整个求虚树的复杂度应该为 $O(n\log n)$。
整体思路为我们用首先把所有关键点按照 dfs 序排序,然后首先把 $1$ 节点加入栈。整个过程我们维护一条左链。我们考虑往里面加入一个新的关键点。求一下这个点和栈顶的 lca。
如果这个这个点和整个单调栈中的点都在一个链上,我们就入栈。
否则,我们不停的弹栈,同时链上栈中元素之间的边,接着,如果这个 lca 不在栈中的话,加入这个 lca 并连相应的边,最后加入我们的新关键点。
注意:
- 如果 $1$ 节点也是关键点注意不要重复入栈,判掉。
- 最后栈中元素连边。
代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
| inline void bu(){ top=0; sort(v.begin(),v.end(),cmp);sta[++top]=1;head2[1]=0;tail2=0; for(int now:v){ vis[now]=1; if(now==1) continue; int nl=ge(now,sta[top]); if(nl!=sta[top]){ while(top>=2&&id[nl]<id[sta[top-1]]) Add2(sta[top-1],sta[top]),top--; if(id[nl]>id[sta[top-1]]) head2[nl]=0,Add2(nl,sta[top]),sta[top]=nl; else Add2(nl,sta[top--]); } sta[++top]=now;head2[now]=0; } for(int i=1;i<top;i++) Add2(sta[i],sta[i+1]); }
|
Author:
HYLTianMeng
License:
Copyright (c) 2019 CC-BY-NC-4.0 LICENSE