Leetcode947

一些坐标上有一些石头，每次移除一个石头的时候在同行或者同列上必须有至少一个石头，问我们可以最多移除多少个石头？

 

sol：

并查集，考虑把在同行或同列的放在一起，假设一共有N个石头，我们总是可以移除n-1个石头。

因此并查集的时候加个计数就可以了

 

class Solution {
    int[] f;
    int[] num;

    public int removeStones(int[][] stones) {
        f = new int[stones.length];
        num = new int[stones.length];
        for (int i = 0; i < f.length; i++) {
            f[i] = i;
            num[i] = 1;
        }
        for (int i = 0; i < f.length; i++) f[i] = i;
        for (int i = 0; i < stones.length; i++) {
            for (int j = i + 1; j < stones.length; j++) {
                boolean can = check(stones[i], stones[j]);
                if (can) {
                    union(i, j);
                }
            }
        }
        int[] cnt = new int[stones.length];

        for (int i = 0; i < num.length; i++) {
            int parent = find(i);
            cnt[parent] = num[parent];
        }
        int res = 0;
        for (int i = 0; i < num.length; i++){
            if (cnt[i] >= 1) res += cnt[i] - 1;
        }
        return res;
    }

    int find(int x) {
        if (x == f[x]) return x;
        else return f[x] = find(f[x]);
    }

    void union(int a, int b) {
        int fa = find(a);
        int fb = find(b);
        if (fa != fb) {
            f[fa] = fb;
            num[fb] += num[fa];
        }
    }

    private boolean check(int[] a, int[] b) {
        return a[0] == b[0] || a[1] == b[1];
    }
}