一道经典题的算法证明

Tree I
题目：
给定一个无向带权连通图，每条边是黑色或白色。试求恰好有条$k$黑色边的生成树的最小权值和。(By WJMZBMR)

一些记号
$f(x):=$恰好有$x$条黑色边的生成树的最小权值和
$S(x):=$所有黑色边权值减去$x$之后的MST的值
$f$是定义在$\mathbb{N}$上的函数，但我们可以对其拓展，$\forall x\in \mathbb{N}$,将$(x,f(x))$与$(x+1,f(x))$相连，从而将$f$拓展到$[0,+\infty ]$。

凸函数性质讨论
直觉上，$f$应该是凸函数，因此先探讨下假如$f$是凸函数时具有哪些性质。
将所有$(x,f(x))$用直线连接起来，可以得到一个凸包。上下凸包只需要以$x$轴为对称轴对称即可互相转换，不失一般性，下述只讨论下凸包的情况。
如下图，任取一条直线$y=kx+b$，假设直线与凸包交于点$(x_0,f(x_0))$，把点代入直线可得截距$b=f(x_0)-k{x}_0$。用这条直线去切这个凸包，一定可以切到一个最优解的点(即一个拐点)，并且这时候截距$b$取最小值，即函数$g(x)=f(x)-kx$取最小值。
注意，直线切到的可能是多个最优解的点，例如下图$A,B,C$三点共线的情况，以斜率为$k_{AC}$的直线去切凸包，会与这三点重合。

那么，对一个给定的$k$，若求出$f(x)-kx$在$[x_1,x_2]$处取得最小值$g(x_1)$，则$[x_1,x_2]$处的最优解$f(x)=g(x)+kx,x\in [x_1,x_2]$。
再考虑上图，由于是一个下凸包，所以对每一个最优解的点，都存在满足上述性质的直线，并且直线的斜率是随着$x$递增的。因此若要求$x_0$处的最优解，只要二分斜率 $k$，解出$g(x)$取最小值时的解集，直到这个解集包含$x_0$，即可求出最优解$f(x_0)$。
事实上，按照上述讨论，有可能不存在整数$k$，使得解集内包含$x_0$，那二分的斜率就必须为浮点数。但是该题的情况比较特殊，如果$f(x_0)=f(x_0-1)=f(x_0+1)$，则用$k=0$的直线就能切到，否则若$f(x_0)\ne f(x_0+1)$且$f(x_0)\ne f(x_0-1)$，由于边的权值都是整数，只要纵坐标不同，则其差至少为1，即$|k_1|=|f(x_0)-f(x_0-1)|\ge 1$且$|k_2|=|f(x_0)-f(x_0+1)|\ge 1$,以及$|k_1-k_2|=|f(x_0-1)-f(x_0+1)|\ge 2$。因此在$k_1,k_2$之间必然存在至少一个整数，$k$取这个整数时即可在$x_0$处与这个凸包相切。
从而在该问题中，只需要二分整数斜率即可。

证明$f$是凸函数
先探讨一下$S(x)$的一些性质。假设$S(k)$对应的生成树选了$x$条黑边，则这个边集在原图上的权值和为$S(k)+kx$，不难发现这个形式与上述的$f(x)$和$g(x)$的关系具有相似性，因此不妨猜想，若$S(k)$对应的生成树选了$x$条黑边，则$f(x)=kx+S(k)$。
下面来证明这个猜想：
由于$S(k)$对应的生成树也是原图的一个生成树，因此$f(x)\le kx+S(k)$。假设$f(x)<kx+S(k)$，则存在一个$x$条黑边的生成树，其权值和为$f(x)$，将其边权减去$k$之后，权值和为$f(x)-kx<S(k)$，这与$S(k)$的定义矛盾，故$f(x)=kx+S(k)$。
对给定的$k$，取满足上式的最小和最大的$x$分别为$l_k,r_k$。接着证明，若$k_1<k_2$，则$r_{k_1}\le l_{k_2}$。
假设命题不成立，对$k_1<k_2$，存在$x_1>x_2$，使得$f(x_1)-k_1{x}_1=S(k_1),f(x_2)-k_2{x}_2=S(k_2)$，则取$k_1,x_1$时的生成树在黑边减去$k_2$情况下的权值和$f(x_1)-k_2{x}_1<S(k_2)$，这与$S(k)$的定义矛盾，由反证法原理，命题成立。
接着可以猜想给定 $k$后，得到 $l_k,r_k$，那么对任意$x\in [l_k,r_k]$，都有$f(x)-kx=S(k)$，即把所有黑边权值减去$k$之后，必存在一棵MST，其黑边数量为$x$。
为了证明这个猜想，先要引入两个MST的性质。
①一张无向图上的所有MST的边权集合是相同的。直观的来说，把MST的边按权值分类，每一类的个数是固定的。
②一张无向图上的不同MST中，某一种权值的边全部连接完后，形成的连通块的连通性是一样的。
因此，由性质②，我们考虑会被选中的权值$p$，先删去所有权值为$p$的边，对得到的图进行缩点，再恢复所有删去的边，此时这张图的边权都是相同的，由性质①，这张图任选一个生成树，都是可以构成原图MST的边集。
假设在新的图的所有生成树中，黑边数量最少为$l^{\prime }$，最多为$r^{\prime }$。我们对此图进行归纳假设。假设存在黑边数量为$x$的一个生成树，考虑任意加一条未选中的黑边，此时会形成一个唯一的环，若这个环中存在一条白边，那么将这条白边删去，即可得到一个黑边数量为$x+1$的生成树。若对所有黑边数量为$x$的生成树，任选未选中的黑边，所在的环都是全部都是黑边的环，就说明选择的黑边已达数量上限，即$x=r^{\prime }$。因此$\forall x\in [l^{\prime },r^{\prime }]$，都存在黑边数量为$x$的生成树。
由于性质②，只要选择的是一个MST的边集的子集，则不同边权的边的选择不会互相影响，假设将不同边权的边分类为边集$E_1,E_2,...,E_s$，则对每类边集$E_i$，都有一个黑边可选的区间$[l_i^{\prime },r_i^{\prime }]$，可以任选$x\in [l_i^{\prime },r_i^{\prime }]$，因此$l_k={\sum }_{i=1}^s{l}_i^{\prime },r_k={\sum }_{i=1}^s{r}_i^{\prime }$，且其中的任何数都可选。
之后，我们再考虑是否所有$x$都被上述性质覆盖了。先证明$S$是一个连续函数。
设N为节点数，$\forall ϵ>0$，取$\delta =\frac{ϵ}{N}$，对任意$|k_1-k_2|<\delta$，$|S(k_1)-S(k_2)|<N\frac{ϵ}{N}=ϵ$，从而 $S$一致连续。
假设存在$x_0$，不存在$k$使得$f(x_0)-k{x}_0=S(k)$。设$x<x_0$且存在$k$，使得$f(x)-kx=S(k)$的最大的$x$为$x_1$。设满足$f(x_1)-k{x}_1=S(k)$的最大的$k$为$k_1$，取$k_2>k_1$，设满足$f(x)-k_{2}x=S(k_2)$的最小的$x$为$x_2$，有$x_1<x_0<x_2$，且$(f(x_1)-f(x_2))-(k_1{x}_1-k_2{x}_2)=S(k_1)-S(k_2)$。令$k_2\to k_1$，由$S$的连续性，$f(x_2)-k_1{x}_2=f(x_1)-k_1{x}_1=S(k_1)$，这与$x_1$的定义不符，从而对任意$x$，都存在 $k$使得$f(x)-kx=S(k)$。
综上，这些性质覆盖了所有的$x$。
整理上述所有结论，由于$f$是连续的，且由上述讨论，可以被等效成一系列斜率递增的线段的并，因此$f$是凸函数。