将方案数和距离和分开计算,方案数用容斥算,距离和就枚举树上的每条边,讨论将3个人分成两边的3种情况分别计算方案数,这条边对距离和的贡献为这条边的长度乘以总方案数,最后期望为距离和除以方案数,时空复杂度O(n)。

相关推荐

牛客网
牛客企业服务