leetcode-947. 移除最多的同行或同列石头-C语言

/*
 * 算法思想：并查集
 * 1. 遍历数组，边遍历，边和前边的元素比较；
 * 2. 如果属于同一类（x或者y相等），则将当前元素归为前面对应元素的一类；
 * 3. 查看并查集中有几类元素，共有cnt类元素；
 * 4. 结果为总元素数-cnt，因为每类最后一个没法移除。
 */

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 removeStones(int** s, int slen, int* stonesColSize){
    int arr[slen+1], i, j, cnt=0;
    int last = INT_MAX;
    
    for(i=0; i<slen+1; i++){
        arr[i] = i;
    }
    
    for(i=0; i<slen; i++){
        for(j=0; j<i; j++){
            if(s[i][0] == s[j][0] || s[i][1] == s[j][1]) {
                un(arr, j+1, i+1);
            }
        }
    }
    
     for(i=1; i<slen+1; i++){
         if(arr[i] != i){
             arr[i] = find(arr, arr[i]);
         }
     }
 
    qsort(arr+1, slen, sizeof(int), cmp);

    for(i=1; i<slen+1; i++){
        if(last != arr[i]){
            last = arr[i];
            cnt++;
        }
    }
   
    
    return slen - cnt;
    
}