leetcode 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 。

示例 1：

输入：n = 4, edges = [[3,1,2],[3,2,3],[1,1,3],[1,2,4],[1,1,2],[2,3,4]]
输出：2
解释：如果删除 [1,1,2] 和 [1,1,3] 这两条边，Alice 和 Bob 仍然可以完全遍历这个图。再删除任何其他的边都无法保证图可以完全遍历。所以可以删除的最大边数是 2 。

代码

class Solution {
    int[] fa;

    public void  init()
    {
        for(int i=0;i<fa.length;i++)
            fa[i]=i;

    }
    public int  find(int x)
    {
        if(x!=fa[x])
          fa[x]=find(fa[x]);
        return fa[x];
    }
    public void   union(int x,int y)
    {
        x=find(x);
        y=find(y);
        if(x==y) return;
        fa[x]=y;

    }

    public int maxNumEdgesToRemove(int n, int[][] edges) {

        int n1=n,n2=n,res=0;
        fa=new int[n+1];
        init();
        for(int[] c:edges)//先把公共边放进去
        {
            if(c[0]==3)
            {
                if(find(c[1])==find(c[2]))
                {
                    res++;
                }else{
                    union(c[1],c[2]);n1--;n2--;
                }

            }
        }
        int[] temp=fa.clone();
        for(int[] c:edges)//在遍历完公关边的基础上，遍历类型为1的所有边
        {
            if(c[0]==1)
            {
                if(find(c[1])==find(c[2]))
                {
                    res++;
                }else{
                    union(c[1],c[2]);n1--;
                }

            }
        }
        fa=temp;
        for(int[] c:edges)//在遍历完公关边的基础上，遍历类型为2的所有边
        {
            if(c[0]==2)
            {
                if(find(c[1])==find(c[2]))
                {
                    res++;
                }else{
                    union(c[1],c[2]);n2--;
                }

            }
        }  
        if(n1!=1||n2!=1)//如果alice和bob其中一个不能遍历所以节点，则返回-1
            return -1;
        else return res;


    }
}