题目:
同样是岛的问题,但是参数有所变化,一共3个参数,m、n、int[][] position。根据position,求出每一步的岛屿的数量。
代表的意思是:m * n是二维数组的行和列,通过 m * n可以构建一个值都为0的二维数组。
而position的值代表着m,n构建的二维数组中对应下标处的值改为1。position[i][j] 代表着mn数组的行和列
举例:
如图所示:m = 3,n = 4,通过m和n构建出来一个3行4列的值都为0的数组。
position的值为[0][0],[0][1],[1][1],mn数组根据position动态改变了,根据position的不同,求出每一步的岛屿的数量。
并查集解法
岛屿数量解法一的帖子中介绍过感染的解法,但是这道题用感染的解法是做不出来的。如果第一次将值“”感染“”后,下一次position的值在上下左右,合并不了了,会变成2个或多个岛屿。
所以这道题只能用并查集的方式来解答,并且和前两篇帖子的解答方式也不大相同。因为position是动态的。
分析
- 同样采用二维数组拉伸为一维数组的方式,通过mn来确定一维数组的长度。
- 不进行初始化赋值操作,因为position是动态的,不知道position中的值都有什么,所以不进行初始化操作。
- 利用size变量,通过size变量的值,来确定该位置是否经历过初始化,如果没有,则进行对应位置的初始化操作,并union上下左右,看是否有“1”可以连成一篇岛屿。
public static List<Integer> numIslands21(int m, int n, int[][] positions) {
UnionFind uf = new UnionFind(m,n);
List<Integer> ans = new ArrayList<>();
for (int[] position : positions){
ans.add(uf.connect(position[0],position[1]));
}
return ans;
}
public static class UnionFind {
int[] parent;
int[] size;
int[] help;
final int col;
final int row;
int sets;
public UnionFind(int m, int n) {
int len = m * n;
parent = new int[len];
size = new int[len];
help = new int[len];
col = n;
row = m;
sets = 0;
}
public int index(int r, int c) {
return r * col + c;
}
public int findParent(int index) {
int num = 0;
while (index != parent[index]) {
help[num++] = index;
index = parent[index];
}
for (int i = 0; i < num; i++) {
parent[help[i]] = index;
}
return index;
}
public int connect(int r, int c) {
int index = index(r, c);
if (size[index] == 0) {
parent[index] = index;
size[index] = 1;
sets++;
union(r, c, r - 1, c);
union(r, c, r + 1, c);
union(r, c, r, c - 1);
union(r, c, r, c + 1);
}
return sets;
}
public void union(int r1, int c1, int r2, int c2) {
if (r1 < 0 || r1 == row || r2 < 0 || r2 == row || c1 < 0 || c1 < col || c2 < 0 || c2 < col) {
return;
}
int i1 = index(r1, c1);
int i2 = index(r2, c2);
//如果size[i] == 0,说明该位置没初始化过。
if (size[i1] == 0 || size[i2] == 0) {
return;
}
int f1 = findParent(i1);
int f2 = findParent(i2);
if (f1 != f2) {
if (size[i1] >= size[i2]) {
size[i1] += size[i2];
parent[i2] = f1;
} else {
size[i2] += size[i1];
parent[i1] = f2;
}
sets--;
}
}
}