leetcode-684.冗余连接-C语言

/**
 * 算法思想： 并查集
 */

int find(int *arr , int i){
    return i==arr[i] ? i : (arr[i] = find(arr, arr[i]));
}

void un(int *arr, int i, int j){
    int x = find(arr, i);
    int y = find(arr, j);
    
    arr[y] = x;
}

#define LEN 0xffff

int* findRedundantConnection(int** eds, int len, 
                             int* edgesColSize, int* returnSize){
    int i, j, arr[LEN], *ret=(int*)malloc(sizeof(int) * 2), max=0;
    
    for(i=1; i<LEN; i++){
        arr[i] = i;
    }
    
    for(i=0; i<len; i++){
        if(find(arr, eds[i][0]) == find(arr, eds[i][1])) {
            ret[0] = eds[i][0];
            ret[1] = eds[i][1];
            break;
        }
        un(arr, eds[i][0], eds[i][1]);
    }

    *returnSize = 2;
    return ret;
}