点此看题面 大致题意: 给你一棵树,每条边为黑色或红色, 求有多少个三元组\((x,y,z)\),使得路径\((x,y),(x,z),(y,z)\)上都存在至少一条红色边. 容斥 我们可以借助容斥思想,用总方案数减去不合法方案数,就可以得到合法方案数. 一个不合法方案,就要使得路径\((x,y),(x,z),(y,z)\)中,至少存在一条路径是全黑的. 如果我们删去树上的红色边,只留下黑色的边.则可以发现,一个不合法方案,满足至少存在两个点在同一个连通块内. 计算答案 考虑用并查集,统计每一个连…