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 。
解题思路：优先连接公共边，即Alice和bob都可以遍历的边，然后再连接类型一和类型二的边，利用并查集的思路，如果当前两个顶点已经属于同一个集合那么这条边就是多余的，可以删去，代码如下

class UnionFind():
    
    def __init__(self, n):
        self.f = list(range(n))
        self.n = n
    def find(self, x):
        root = x
        while(root != self.f[root]):
            root = self.f[root]
        while(x != root):
            origin_r = self.f[x]
            self.f[x] = root
            x = origin_r
        return root
    def union(self, x, y):
        rx, ry = self.find(x), self.find(y)
        if rx == ry:
            return True
        self.f[ry] = rx
        return False
    
    def getCount(self):
        for i in range(1, self.n):
            self.find(i)
        return len(set(self.f[1:]))
class Solution:
    def maxNumEdgesToRemove(self, n: int, edges: List[List[int]]) -> int:
        alice = UnionFind(n+1)
        bob = UnionFind(n+1)
        type3s, typeothers = [], []
        res = 0
        for t, u, v in edges:
            if t == 3:
                if alice.union(u, v):
                    res += 1
                else:
                    bob.union(u, v)
            else:
                typeothers.append([t, u, v])
                
        for t, u, v in typeothers:
            if t == 1:
                if alice.union(u, v):
                    res += 1
            elif t == 2:
                if bob.union(u, v):
                    res += 1
        flag = alice.getCount() == 1 and bob.getCount() == 1
        return res if flag else -1