题目大意
给定一棵无根树,函数S(x,y)=∑v∈V(W(v)∗min(d(x,v),d(y,v)))S(x,y)=\sum\limits_{v\in V}(W(v)*min(d(x,v),d(y,v)))S(x,y)=v∈V∑(W(v)∗min(d(x,v),d(y,v))),其中VVV是点集,W(v)W(v)W(v)是vvv点的权值,d(x,y)d(x,y)d(x,y)表示树上两点之间的距离。
求当xxx和yyy取多少时,S(x,y)S(x,y)S(x,y)能取到最小值,并输出这个最小值。
Solution
首先需要统一的是,W(v)W(v)W(v)是一定的,所以如果只有111个点时,就必定取重心。
然而题目中是222个点,但是S(x,y)S(x,y)S(x,y)的含义还是和重心所差无几。一旦选中了x,yx,yx,y,那么整棵树的点集就分成了到xxx点和到yyy点两部分,而这两部分一定是内部互相连通的,所以开始的时候不妨把它看做是切成222棵子树,然后对于每棵子树都求一下重心,作为x,yx,yx,y。
读题后发现,树的高度最大是100100100,利用这一特性,如果我们可以在O(h)O(h)O(h)的范围内求解任意子树的重心,那么这题就得解了。我们可以枚举断边,然后对于每次断边,分别求出两棵子树的重心,然后更新答案。
-
那么可以在O(h)O(h)O(h)内求解重心吗?
-
可以的!
首先设f(u)f(u)f(u)表示整棵树中所有节点到uuu的距离和(当然要乘上点权),siz[u]siz[u]siz