2020.07.21日常总结——换根dp略讲

$$\text{换根dp}$$

严格来说，其实并没有 换根dp 这一种 dp 方法，它只是 树上dp 问题的一个解题技巧。但是由于这个技巧太过于重要了，所以我们可以把它叫做一个单独类型的 dp 方法。

所谓的 换根dp，是针对无根树上 dp 问题的一个解题技巧。

众所周知，我们解无根树上 dp 问题时一般会先强制一个节点（一般来说，为了方便，会选定 $1$ 号点）当整棵树的根，然后在考虑求解。

但是，如果要求出所有点做根时的 dp 值该怎么办呢？暴力 $O(n)$ 枚举？不是不行，但是在大多数时候会 TLE（注意，我没有说一定不行！）。

dp 题的一大特色是：理论非常难讲懂，所以我们直接上一道例题。

---

$$\text{P2986 [USACO10MAR]Great Cow Gathering G}$$

$\text{[Problem]}$

$\text{[solution]}$

最原始的想法是一个 $O(n^2)$ 的暴力：$O(n)$ 枚举节点，$O(n)$ 求出所有牛到一个节点的不方便值。

可是 $1\le n\le 1\times 10^5$，$O(n^2)$ 这样的算法实在是太慢了。有没有快点的方法？

有。

我们记 $f_u$ 表示所有牛到 $u$ 这个节点的不方便值，则：

$$f_u=\sum _{i=1}^n{C}_i\times s_{i,u}$$

其中，$s_{i,u}$ 表示 $i$ 和 $u$ 之间的距离。

我们再记 $s{z}_u$ 表示 $u$ 的子树的大小（注意：这个大小指的是有多少头牛而非有多少个节点）。

显然，我们会先指定一个节点当根（这个根就是所有奶牛要到的节点），我们不妨就指定 $1$ 吧，$O(n)$ 递归求出 $s{z}_i(1\le i\le n)$ 和 $f_1$。

如何用 $f_1$ 和 $sz$ 推导出其它的 $f_i(2\le i\le n)$ 呢？我们发现这个移动节点的过程就是无根树移动根的过程，我们可以画这么一幅图：

这就是一个换根的过程。

它能告诉我们一些什么呢？首先我们发现以 $3$ 为根的子树内的所有的点需要做得距离少了 $s_{1,3}$，但那些不是 $3$ 的儿子的点（包括 $1$），就需要多走 $s_{1,3}$ 的距离。

所以，在这幅图中，我们有：

$$f_3=f_1-s{z}_3\times s_{1,3}+(n-s{z}_3)\times s_{1,3}$$

类似地，我们可以推广到一般情况：

$$f_v=f_u-s{z}_v\times s_{u,v}+(n-s{z}_v)\times s_{u,v}$$

于是，我们再 dfs 一次，就可以在 $O(n)$ 的时间复杂度内求出所有的 $f_i(1\le i\le n)$ 了。

总时间复杂度：$O(n)$。

$\text{[code]}$

const int N=1e5+100;
struct edge{//链式前向星 
	int next,to,len;
}e[N<<1];int h[N],tot;
inline void add(int a,int b,int c){
	e[++tot]=(edge){h[a],b,c};h[a]=tot;
	e[++tot]=(edge){h[b],a,c};h[b]=tot;
}
int sze[N],n,C[N],cnt;//含义如题 
long long F1,ans;//要开long long 
typedef long long ll;//为了方便 
void dfs1(int u,int fa,ll s){
	F1+=1ll*C[u]*s;sze[u]=C[u];
	for(int i=h[u];i;i=e[i].next){
		register int to=e[i].to;
		if (to==fa) continue;//!
		dfs1(to,u,s+e[i].len);
		sze[u]+=sze[to];//updata
	}
}
void dfs2(int u,int fa,ll Fu){
	for(int i=h[u];i;i=e[i].next){
		register int to=e[i].to;
		if (to==fa) continue;//!
		ll Fv=Fu-1ll*sze[to]*e[i].len
		      +1ll*(cnt-sze[to])*e[i].len;
		ans=min(ans,Fv);//更新答案 
		dfs2(to,u,Fv);//继续递归 
	}
}
int main(){
	n=read();F1=0;//init 
	for(int i=1;i<=n;i++)
		cnt+=(C[i]=read());
	for(int i=2;i<=n;i++){
		int u=read(),v=read(),w=read();
		add(u,v,w);//按照题目要求加边 
	}
	dfs1(1,-1,0);//求F1 
	ans=F1;//ans最大为F1 
	dfs2(1,-1,F1);//换根 
	printf("%lld",ans);
} 

---

$$\text{总结}$$

• 一般地，做这种题目时，步骤如下：

  • 选定一个节点当根，求出它的 dp 值。
  • 考虑从它转移到其它节点的变化量。
  • 两遍 dfs 求解。

• 参考：https://sflsrick.blog.luogu.org/note-how-to-change-root

• 另一篇类似的我的博客：https://blog.csdn.net/zhuyingye_123456/article/details/105026661#comments