n 块石头放置在二维平面中的一些整数坐标点上。每个坐标点上最多只能有一块石头。
如果一块石头的 同行或者同列 上有其他石头存在,那么就可以移除这块石头。
给你一个长度为 n 的数组 stones ,其中 stones[i] = [xi, yi] 表示第 i 块石头的位置,返回 可以移除的石子 的最大数量。
示例 1:
输入:stones = [[0,0],[0,1],[1,0],[1,2],[2,1],[2,2]]
输出:5
解释:一种移除 5 块石头的方法如下所示:
1. 移除石头 [2,2] ,因为它和 [2,1] 同行。
2. 移除石头 [2,1] ,因为它和 [0,1] 同列。
3. 移除石头 [1,2] ,因为它和 [1,0] 同行。
4. 移除石头 [1,0] ,因为它和 [0,0] 同列。
5. 移除石头 [0,1] ,因为它和 [0,0] 同行。
石头 [0,0] 不能移除,因为它没有与另一块石头同行/列。
示例 2:
输入:stones = [[0,0],[0,2],[1,1],[2,0],[2,2]]
输出:3
解释:一种移除 3 块石头的方法如下所示:
1. 移除石头 [2,2] ,因为它和 [2,0] 同行。
2. 移除石头 [2,0] ,因为它和 [0,0] 同列。
3. 移除石头 [0,2] ,因为它和 [0,0] 同行。
石头 [0,0] 和 [1,1] 不能移除,因为它们没有与另一块石头同行/列。
示例 3:
输入:stones = [[0,0]]
输出:0
解释:[0,0] 是平面上唯一一块石头,所以不可以移除它。
提示:
1 <= stones.length <= 1000
0 <= xi, yi <= 104
不会有两块石头放在同一个坐标点上
基本思路:首先,要明白不同的移除步骤,结果可能不同,就以例1为例,可能出现只移除4个石头的结果。那么怎么移除可以移除的石头结果最多,若采用深度或者广度优先遍历,若不是尾节点,可以删除,最后会发现同一个连通域的可以只剩下一个石头。因为问题转换成求连通域的问题,采用并查集
- 需要注意的是,为了防止[1,2],[2,1],这种被误认为是连通的,需要把x或者y坐标特殊处理,本题中,y=y+10001;
- 关于计算顶点的个数,注意[0,0]不是一个点,而是两个点[0,10001].
int count=0;
int n=20005;
vector<int> parent,psize;
unordered_set<int> ss;
int find(int x){
if(ss.count(x)==0){
count++;
}
ss.insert(x);
if(x!=parent[x]){
parent[x]=find(parent[x]);
}
return parent[x];
}
void connect(int x,int y){
int root_x=find(x);
int root_y=find(y);
if(root_x==root_y)
return;
if(psize[root_x]>psize[root_y]){
parent[root_y]=root_x;
psize[root_x]+=psize[root_y];
}
else{
parent[root_x]=root_y;
psize[root_y]+=psize[root_x];
}
count--;
}
void init(int len){
psize.resize(n,1);
parent.resize(n,0);
for(int i=0;i<n;i++){
parent[i]=i;
}
}
int removeStones(vector<vector<int>>& stones) {
int len=stones.size();
init(len);
for(auto &p:stones){
connect(p[0],p[1]+10001); //防止[1,2]和[2,1],连在一起
}
return len-count;
}