题目描述:
在二维平面上,我们将石头放置在一些整数坐标点上。每个坐标点上最多只能有一块石头。
现在,move 操作将会移除与网格上的某一块石头共享一列或一行的一块石头。
我们最多能执行多少次 move 操作?
示例 1:
输入:stones = [[0,0],[0,1],[1,0],[1,2],[2,1],[2,2]]
输出:5
示例 2:
输入:stones = [[0,0],[0,2],[1,1],[2,0],[2,2]]
输出:3
示例 3:
输入:stones = [[0,0]]
输出:0
提示:
1 <= stones.length <= 1000
0 <= stones[i][j] < 10000
很明显使用的是并查集来进行,和之前的朋友圈和等式成立的基本条件是一致的,这里的条件是横坐标或者是纵坐标一致就合并,合并的时候注意:这里根据索引来添加。
代码:
class Solution {
public int removeStones(int[][] stones) {
if(stones.length <= 1) {
return 0;
}
//使用并查集
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < stones.length; i++) {
for (int j = i + 1; j < stones.length; j++) {
if(map.get(i) == null){
map.put(i, i);
}
if(map.get(j) == null){
map.put(j, j);
}
if(stones[i][0] == stones[j][0] || stones[i][1] == stones[j][1]){
Unio(j,Find(i, map),map);
}
}
}
Set<Integer> set = new HashSet<>();
for (int i = 0; i < stones.length; i++) {
set.add(Find(i, map));
}
System.out.println(map.toString());
return stones.length - set.size();
}
public static void Unio(int i,int j,Map<Integer, Integer> map){
map.put(Find(i, map), j);
}
public static int Find(int i,Map<Integer, Integer> map){
return map.get(i) == i? i:Find(map.get(i), map);
}
}
排名靠前的代码
class Solution {
public int removeStones(int[][] stones) {
int len=stones.length;
DSU dsu=new DSU(20000);
for(int i = 0; i < len; i++){
dsu.union(stones[i][0], stones[i][1] + 10000);
}
Set<Integer> set = new HashSet();
//查看连通子图的数量
for(int[] stone : stones){
set.add(dsu.find(stone[0]));
}
return len - set.size();
}
}
//构建并查集类
class DSU{
int pre[];
public DSU(int n){
pre = new int[n];
for(int i = 0; i < n; i++){
pre[i] = i;
}
}
public int find(int x){
while(pre[x] != x){
x = pre[x];
}
return x;
}
public void union(int x, int y){
int fx = find(x);
int fy = find(y);
if(fx != fy){
pre[fx] = fy;
}
}
}
思路基本一致