LeetCode 685. 冗余连接 I--bfs

685. 冗余连接 II

在本问题中，有根树指满足以下条件的 有向 图。该树只有一个根节点，所有其他节点都是该根节点的后继。该树除了根节点之外的每一个节点都有且只有一个父节点，而根节点没有父节点。

输入一个有向图，该图由一个有着 n 个节点（节点值不重复，从 1 到 n）的树及一条附加的有向边构成。附加的边包含在 1 到 n 中的两个不同顶点间，这条附加的边不属于树中已存在的边。

结果图是一个以边组成的二维数组 edges 。 每个元素是一对 [ui, vi]，用以表示 有向 图中连接顶点 ui 和顶点 vi 的边，其中 ui 是 vi 的一个父节点。

返回一条能删除的边，使得剩下的图是有 n 个节点的有根树。若有多个答案，返回最后出现在给定二维数组的答案。

示例 1：

输入：edges = [[1,2],[1,3],[2,3]]
输出：[2,3]

示例 2：

输入：edges = [[1,2],[2,3],[3,4],[4,1],[1,5]]
输出：[4,1]

提示：

n == edges.length
3 <= n <= 1000
edges[i].length == 2
1 <= ui, vi <= n

题解

比较简单，因为数据量比较小。直接暴力BFS搜索就行。

AC代码

class Solution {
public:
    vector<int>to[1010];
    queue<int>q;
    bool vis[1010];
    bool bfs(vector<vector<int>>& edges,int edge_u,int edge_v)
    {
        memset(vis,0,sizeof(vis));
        for(int i=0;i<edges.size();i++)
        {
            if(edges[i][0]==edge_u&&edges[i][1]==edge_v)continue;
            vis[edges[i][1]]=true;
        }
        int x,ans=0;
        for(int i=1;i<=edges.size();i++)
        {
            if(vis[i]==false)
            {
                x=i;
                ans++;
            }
        }
        if(ans!=1)return false;//根节点不唯一说明边删除错了
        memset(vis,0,sizeof(vis));
        while(!q.empty())q.pop();
        vis[x]=true;
        q.push(x);
        ans=0;
        while(!q.empty())
        {
            int u=q.front();
            q.pop();
            ans++;
            for(int i=0;i<to[u].size();i++)
            {
                int v=to[u][i];
                if(vis[v])continue;
                if(u==edge_u&&v==edge_v) continue;//当前此路不通
                vis[v]=true;
                q.push(v);
            }
        }
        //cout<<x<<" "<<ans<<endl;
        return ans == edges.size();
    }
    vector<int> findRedundantDirectedConnection(vector<vector<int>>& edges) 
    {
        for(int i=0;i<edges.size();i++)
        {
            to[edges[i][0]].push_back(edges[i][1]);
        }
        vector<int>res;
        for(int i=edges.size()-1;i>=0;i--)
        {
            if(bfs(edges,edges[i][0],edges[i][1]))
            return edges[i];
        }
        return res;
    }
};