leetcode-684冗余连接

在本问题中, 树指的是一个连通且无环的无向图。

输入一个图，该图由一个有着N个节点 (节点值不重复1, 2, ..., N) 的树及一条附加的边构成。附加的边的两个顶点包含在1到N中间，这条附加的边不属于树中已存在的边。

结果图是一个以边组成的二维数组。每一个边的元素是一对[u, v] ，满足 u < v，表示连接顶点u 和v的无向图的边。

返回一条可以删去的边，使得结果图是一个有着N个节点的树。如果有多个答案，则返回二维数组中最后出现的边。答案边 [u, v] 应满足相同的格式 u < v。

示例 1：

输入: [[1,2], [1,3], [2,3]]
输出: [2,3]
解释: 给定的无向图为:
  1
 / \
2 - 3

示例 2：

输入: [[1,2], [2,3], [3,4], [1,4], [1,5]]
输出: [1,4]
解释: 给定的无向图为:
5 - 1 - 2
    |   |
    4 - 3

注意:

• 输入的二维数组大小在 3 到 1000。
• 二维数组中的整数在1到N之间，其中N是输入数组的大小。

class Solution {
public:
    /*
    分析：
    1、n个顶点的最小生成树一定有n-1条边，若加1条树中不存在的条边则必构成环。
    2、从前向后遍历边集合，若两顶点早已连通，则该边是多余的。
    3、从前向后遍历遇到的第一个多余边就是答案吗？是的，反证法可证。假设此边不是答案，那么要删除后面的某一条边，也就是要保留此边。此边与其前面的边仍可构成环。与题目要求矛盾，故假设不成立，此边是所求答案。
    解法：并查集
    */
    
    int _find(int *s, int x)
    {
        return s[x]-x?s[x]=_find(s,s[x]):x;
    }
    vector<int> findRedundantConnection(vector<vector<int>>& edges) {
        int n = edges.size();
        int *s = new int[n+1];
        for(int i=1;i<=n;i++)
            s[i] = i;
        for(int i=0;i<n;i++)
        {
            int a = _find(s,edges[i][0]), b = _find(s,edges[i][1]);
            if(a==b)
            {
                delete[] s;
                return edges[i];
            }
            s[a] = b;
        }
        return edges[n-1];
    }
};