947. 移除最多的同行或同列石头

947. 移除最多的同行或同列石头

n 块石头放置在二维平面中的一些整数坐标点上。每个坐标点上最多只能有一块石头。

如果一块石头的 同行或者同列 上有其他石头存在，那么就可以移除这块石头。

给你一个长度为 n 的数组 stones ，其中 stones[i] = [xi, yi] 表示第 i 块石头的位置，返回 可以移除的石子 的最大数量。

思路

• 可以想到，剩余的石头各自成一个连通分量，同行同列的石头属于一个连通分量，可以移除到只剩一个石头。
• 因此，对二维平面的点进行并查集操作，将x坐标映射到10000+x，y坐标保持不变，将同行同列的石头全部连通起来。
• 每创建一个独立点时，连通分量+1，每连通两个点时，连通分量-1。

代码

class Solution {
    int count = 0;
    unordered_map<int, int> parent;
public:
    int find(int x) {
        if(!parent.count(x)) {
            parent[x] = x;
            count++;
        }
        if(parent[x] != x) {
            parent[x] = find(parent[x]);
        }
        return parent[x];
    }
    void join(int x, int y) {
        int f_x = find(x);
        int f_y = find(y);
        if(f_x == f_y) {
            return;
        }
        parent[f_y] = f_x;
        count--;
    }
    int removeStones(vector<vector<int>>& stones) {
        for(auto& stone : stones) {
            join(stone[0] + 10000, stone[1]);
        }
        return stones.size() - count;
    }
};