因为排序的原因，在正确的排序算法下，相同权值的边中，白边下标小，黑边下标大，所以就不可避免的出现无论如何都取不到 $need$ 条白边的情况，这时候，你的最终答案就不能减去你生成树中的白边数量成上 $mid$，应该减去 $need\times mid$，代表多余的白边已经找到相同权值且作用相同的黑边代替。

因为题中保证有解，所以做法是正确的。