每日一题——947. 移除最多的同行或同列石头

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

分类：图、并查集

以前的并查集中，总是元素由一个一元表示，而这次元素由一个二元表示

最初的想法是，把行或列相同的点连接，最后点的总数-集合总数就是最多可移除。但是实际上一些例子是无法通过的。暂时想不通为什么。

参考题解，做法是将每个石头的行和列连接。我觉得可以这么考虑，如果所有的石头都不同行列，那么二元的每一个位置都没有交集，二元的两个位置相连接，一共有多少个石头就有多少个集合。如果两个石头产生交集，二元连接后就只产生一个集合，视为一个元素。

class Solution {
public:
    int root(unordered_map<int, int> &u, int p){
        if(!u.count(p))
            u[p] = p;
        return p == u[p] ? p : u[p] = root(u, u[p]);
    }

    void connect(unordered_map<int, int> &u, int i, int j){
        u[root(u, j)] = root(u, i);
    }

    bool isConnected(unordered_map<int, int> &u, int i, int j){
        return root(u, i) == root(u, j);
    }

    int numberOfConnectedComponent(unordered_map<int, int> &u){
        int num = 0;
        for (auto &[k, v] : u) {
            if (k == v) {
                num++;
            }
        }
        return num;
    }

    int removeStones(vector<vector<int>>& stones) {
        int n = stones.size();
        unordered_map<int, int> u;
        for(int i=0; i<n; i++){
            connect(u, stones[i][0], stones[i][1]+10000);
        }
     
        return n-numberOfConnectedComponent(u);
    }
};

2010/01/15