2020杭电多校第六场 1006 A Very Easy Graph Problem

题目

题目大意：
给定一张无向图，图中点分为黑点和白点。对于第i条边，其边长为2ⁱ，问所有黑白点之间最短路之和。

---

生成树

这个题仔细思考一下，不难发现对于第i条边，有

2¹ + 2² + … + 2^i-1 < 2ⁱ
即前面所有的边权和加起来也没有这一条边这么多，因此当第i条边加入时，可以连通的黑点和白点间产生的路径就是最短路径。如果这条边连接的两个顶点已经连通，则不用加入改点。

分析到这里，前面的部分就可以使用并查集直接维护一颗生成树了。接着这个问题就变成了树上统计问题。

树上统计

生成树过后，问题就转变为在一棵无根树上，所有的黑白点对之间的距离和是多少。

为了解决这个问题，我们可以在每一个节点统计几个值：

• sum,以该节点为根的子树中所有黑白点对的距离和
• dp[0][0]，子树中所有黑节点到该节点的距离和
• dp[0][1]，子树中黑节点的数量
• dp[1][0]，子树中所有白节点到该节点的距离和
• dp[1][1]，子树中白节点的数量
• color,节点颜色

不难得到以下转移

对于节点k,其父节点为f：

子树节点颜色数量统计有：
$$node[k].dp[0][1]=\sum _{e(j,k),j\ne f}node[j].dp[0][1]$$
$$node[k].dp[1][1]=\sum _{e(j,k),j\ne f}node[j].dp[1][1]$$
子树距离和统计有：
$$node[k].dp[0][0]=$$