题解 | #小红的树切割#

小红的树切割

题目要求切割后对于任意一棵树，需要满足树中任意相邻的两个节点的颜色不同。

因此在切割前，对于任意一条树上的边，只要与的颜色相同，就需要进行切割，否则不需要进行切割。

所以我们需要遍历树上的每一条边，检验边的两个端点颜色是否相同。

def solve(testcase):
    n = II()
    s = I()

    res = 0
    for _ in range(n - 1):
        u, v = GMI()
        res += s[u] == s[v]
    
    print(res)

for testcase in range(1):
    solve(testcase)