[学习笔记] 虚树

今天考了个模拟赛，虚树的题我拿(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;
}