leecode 冗余连接

题目

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

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

给定的图是以数组形式给出的，每个数组的长度为2，表示附加的边的两个顶点。

你需要选择一条可以删去的边，返回使得图成为一棵树的边。

如果有多个答案，则返回数组中最后出现的边。答案边的格式为 `[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是输入数组的大小。

思路

这题用并查集，其他dfs这些时间复杂度都是O(边数+节点数)，并查集是O（边数），因为只是数组中的某个边查一次，快一点。

代码

#include <stdio.h>

#define SIZE 1000 
int parent[SIZE];//全局变量便于函数操作

// 并查集的初始化
void initialize(int size) {
    for (int i = 0; i < size; i++) {
        parent[i] = i;  // 最初，每个元素的父节点是它自己
    }
}

int find(int x) //找集合的根
{
    if (parent[x] != x) //如果一个节点的爹不是自己，就说明它不是根节点
    {
        parent[x] = find(parent[x]);  //继续给他找爹的爹看看有没有
    }
    return parent[x];//这里返回的是最终找到没有父节点的根
}

// 合并操作
void union_sets(int x, int y) {
    int rootX = find(x);
    int rootY = find(y);
    if (rootX != rootY) //如果根节点不是同一个，则将y当成x的子节点处理
    {
        parent[rootY] = rootX;  // 将一个集合的根节点连接到另一个集合的根节点
    }
}

// 找到冗余连接
int* findRedundantConnection(int edges[][2], int edgesSize, int* returnSize) {
    initialize(edgesSize + 1);  // 初始化并查集，节点编号从1开始，所以数组大小是edgesSize+1
    for (int i = 0; i < edgesSize; i++) //因为这边需要找到的是最后出现在数组里面的冗余链接，所以我们直接遍历看最后出现的冗余边就行了
    {
        int u = edges[i][0];
        int v = edges[i][1];
        if (find(u) == find(v)) // u，v根相同的话
        {  
            *returnSize = 2;
            return edges[i];
        } 
        else 
        {
            union_sets(u, v);  // 把u,v的爹设置为同一个
        }
    }
    *returnSize = 0;
    return NULL;
}

int main() {
    int edges[][2] = {{1, 2}, {1, 3}, {2, 4}, {2, 5}, {3, 6}, {4, 7}, {3, 7}};
    int edgesSize = sizeof(edges) / sizeof(edges[0]);
    int returnSize;
    int* redundant = findRedundantConnection(edges, edgesSize, &returnSize);

    if (redundant != NULL) {
        printf("Redundant connection: [%d, %d]\n", redundant[0], redundant[1]);
    } else {
        printf("No redundant connection found\n");
    }

    return 0;
}

结果