今天考了个模拟赛,虚树的题我拿(LCT)粗暴卡过,然后被各路神仙疯狂嘲讽,然后就奋发图强,来学了个虚树。
虚树的概念
先放一个例题吧:[SDOI 2011]消耗战
例题单个询问的树形DP可以说是非常简单了,但是多个询问就会(GG),于是我们痛定思痛,发现其实我们并不需要整棵树来转移,我们只需要得到每一个目标节点和他们的(LCA)所构成的树即可,而树的边权即为树上两点在原树上的路径中的最小值。
那么像这种由目标节点和他们的(LCA)所构成的树即为虚树。
虚树的构造
既然知道了啥是虚树,那么就让我们再来看看虚树应该怎么构造吧。
其实虚树的构造过程类似(dfs)。
我们先整个类似(dfs)的栈,这个栈里存的只有虚树上需要的节点。
那么对于一个新加入的目标节点 (x)(显然目标节点要按(dfs)序排序再加入),我们分两种情况讨论:
-
(x) 在以(stk[top])为根的子树中
-
(x) 不在以(stk[top])为根的子树中
第一种情况很简单,直接把 (x) 加入栈即可。
让我们来具体研究一下第二种情况:
我们找到次栈顶结点((stak[top-1]))与 (x) 的 (LCA) ,我们只需要一直把次栈顶与(LCA)比较,若 (dep[stk[top-1]]>dep[LCA]) ,那么便将给栈顶((stk[top]))和次栈顶((stk[top-1]))之间连一条边。
如果某一时刻(dep[stk[top-1]]<dep[LCA]&&dep[stk[top]]==dep[LCA]),那么就说明(LCA)就是(stk[top]),我们只需要连上((stk[top],x))这条边即可。
那么如果(dep[stk[top]]>dep[LCA])呢?
这就说明(LCA)并不在虚树中,我们需要连上((stk[top],LCA)),然后将(LCA)和 (x) 加入栈中。
最后,给出建虚树代码:
void ins(int x){
int l=lca(x,stk[top],0);
while (top>1&&dep[l]<dep[stk[top-1]]){
E[stk[top]].push_back(stk[top-1]);
E[stk[top-1]].push_back(stk[top]);
--top;
}
if (dep[stk[top]]>dep[l]){
E[stk[top]].push_back(l);
E[l].push_back(stk[top]);
--top;
}
if (stk[top]!=l) stk[++top]=l;
stk[++top]=x;
}