力扣刷题记录 -- JAVA---150--685. 冗余连接 II

最新推荐文章于 2024-08-26 22:22:22 发布

@白圭

最新推荐文章于 2024-08-26 22:22:22 发布

阅读量42

点赞数

分类专栏： # 力扣习题记录册__JAVA__代码随想录文章标签： leetcode java 算法

本文链接：https://blog.csdn.net/taiyuezyh/article/details/133325184

版权

力扣习题记录册__JAVA__代码随想录专栏收录该内容

151 篇文章 0 订阅

订阅专栏

一、题目

在这里插入图片描述

二、代码

class Solution 
{
    //并查集的嵌套
    //方案一: 有一个入度为2的节点  删除一条边
    //方案二: 构成环 删除一条边
    int n = 1001;
    int[] father = new int[n];
    public void init()
    {
        int i;
        for(i=0;i<n;i++)
        {
            father[i] = i;
        }
    }
    public int find(int u)
    {
         if(u==father[u])  return u;
         father[u] = find(father[u]);
         return father[u];
    }
    public boolean same(int u,int v)
    {
        u = find(u);
        v = find(v);
        return u==v;
    }
    public void join(int u, int v)
    {
        u = find(u);
        v = find(v);
        if(u==v) return ;
        father[u] = v;

    }
   
   public int[] getRemoveEdge(int[][] edges)
   {
       int i;
       init();

       for(i=0;i<edges.length;i++)
       {
           if(same(edges[i][0],edges[i][1]))
           {
               return edges[i];
           }
           join(edges[i][0],edges[i][1]);
       }
       return null;
   }
   public boolean isTreeAfterRemoveEdge(int[][] edges , int deleteEdge)
   {
       int i;
       init();
       for(i=0 ; i<edges.length;i++)
       {
           if(i == deleteEdge) continue;
           if(same(edges[i][0],edges[i][1]))
           {
               return false;
           }
           join(edges[i][0],edges[i][1]);
       }
       return true;
   }

    public int[] findRedundantDirectedConnection(int[][] edges) 
    {
        int i,j;
        int[] inDegree = new int[n];
        for(i=0;i<edges.length;i++)
        {
            inDegree[edges[i][1]] = inDegree[edges[i][1]] + 1 ;
        }  
        LinkedList<Integer> twoDegree = new LinkedList<Integer>();
        for(i = edges.length - 1 ; i>=0 ;i--)
        {
            if(inDegree[edges[i][1]] == 2)
            {
                twoDegree.add(i);
            }
        }
        
        //存在入度为2  必为其中一条边
        if(!twoDegree.isEmpty())
        {
            if(isTreeAfterRemoveEdge(edges , twoDegree.get(0)))
            {
                return edges[twoDegree.get(0)];
            }
            return edges[twoDegree.get(1)];
        }

        return getRemoveEdge(edges);

        //否则有环
    }
}