leetcode-547. 朋友圈-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;
}

int cmp(int *a, int *b){
    return *a - *b;
}

int findCircleNum(int** M, int m, int* ns){
    int i, j, len = m + 1, n = ns[0], last=INT_MAX, cnt=0;
    if(!m || !m) return 0;
    
    int arr[len];
    
    for(i=1; i<len; i++){
        arr[i] = i;
    }
    
    for(i=0; i<m; i++){
        for(j=0; j<n; j++){
            if(M[i][j]) {
                un(arr, i+1, j+1);
            }
        }
    }
    
    for(i=1; i<len; i++){
        if(arr[i] != i){
            arr[i] = find(arr, arr[i]);
        }
    }
    
    qsort(arr+1, len-1, sizeof(int), cmp);
    
    for(i=1; i<len; i++){
        if(arr[i] != last){
            cnt++;
            last = arr[i];
        }
    }   
    
    return cnt;
}