【1月打卡~Leetcode每日一题】1579. 保证图可完全遍历（难度：困难）

1579. 保证图可完全遍历

Alice 和 Bob 共有一个无向图，其中包含 n 个节点和 3 种类型的边：
类型 1：只能由 Alice 遍历。
类型 2：只能由 Bob 遍历。
类型 3：Alice 和 Bob 都可以遍历。
给你一个数组 edges ，其中 edges[i] = [typei, ui, vi] 表示节点 ui 和 vi 之间存在类型为 typei 的双向边。请你在保证图仍能够被 Alice和 Bob 完全遍历的前提下，找出可以删除的最大边数。如果从任何节点开始，Alice 和 Bob 都可以到达所有其他节点，则认为图是可以完全遍历的。
返回可以删除的最大边数，如果 Alice 和 Bob 无法完全遍历图，则返回 -1 。

思路：并查集问题，考虑以下两种优先级
① 3种类型的边是有优先级
② 同一类型的两条不同的边无优先级

考虑①：Alice可以使用类型1和3，Bob可以使用类型2和3，使用越多的类型3，即公共边，则可以删除更多的边数，因此类型3优先级高
考虑②：边的作用在于连接两个连通分量，那么一条边的作用只能连接2个不同的分量，也就是说两个不同的分量间，使用哪一条边都是可以的，因此两条不同的边无优先级

因此：先单独遍历类型3，再遍历1和2，使用两个数组分别表示Alice和Bob的连通情况，如果一条边连接的两个点不属于同一连通分量，则连接，否则ans + 1

class Solution:
    def maxNumEdgesToRemove(self, n: int, edges: List[List[int]]) -> int:
        parent = list(range(n))
        ans = 0

        def find(index,parent):
            if index != parent[index]:
                parent[index] = find(parent[index],parent)
            return parent[index]

        def union(index1,index2,parent):
            parent[find(index1,parent)] = find(index2,parent)

        for relation in edges:
            x,y = relation[1]-1,relation[2]-1
            type_ = relation[0]
            if type_ == 3:
                if find(x,parent)^find(y,parent):
                    union(x,y,parent)
                else:
                    ans += 1

        parent_alice = [i for i in parent]
        parent_bob = [i for i in parent]
        f = {1:parent_alice,2:parent_bob}

        for relation in edges:
            x,y = relation[1]-1,relation[2]-1
            type_ = relation[0]
            if type_ ^ 3: 
                if find(x,f[type_])^find(y,f[type_]):
                    union(x,y,f[type_])
                else:
                    ans += 1

        return ans if sum([i==parent_alice[i] for i in range(n)]) == sum([i==parent_bob[i] for i in range(n)]) ==1 else -1