洛谷P3478
先来看一道例题,很经典的一道换根DP题。考虑暴力,依次枚举以每一个点为根的深度之和,复杂度是 ,不能通过此题。这时就需要用换根DP把复杂度降为
。
我们先进行一次 ,假设以
为根,把深度之和求出来。
第二次 是换根DP的关键。考虑根节点由
变成
,那么
号节点和
号节点的深度要减一,
号节点的深度要加一。也就是说,
子树内的节点深度都要减一,其余的节点深度都要加一。
于是有转移式 ,即
。
一定要注意,先把以 号节点为根的总深度求出来,即先求出