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