岛问题
一个矩阵中只有0和1两种值,每个位置都可以和自己的上、下、左、右 四个位置相连,如果有一片1连在一起,这个部分叫做一个岛,求一个 矩阵中有多少个岛?
举例:
这个矩阵中有三个岛。
思想:1:使用递归DFS,深度搜索;
2:使用并查集;这里主要说一下使用并查集的方法。
假设把那个大的岛分成两个岛,这样子分两个的时候有4个岛,那么我们可以怎么算出岛的个数呢
** 我们只需要看这两个岛的边界,看边界是被那个集合感染的,然后依次对比,比如看如图的这两个岛的边界,第一个1-1 A和C不是同一个集合,合并后-1。第一个1-1 A和C已经合并了,所以是同一个集合,合并后不-1。第三个0-1不比较。第四个B和C不是同一个集合,合并后-1,;第五个1-0不比较。所有总共的岛的个数为4-2=2。**
为什么使用并查集更好呢,试想一下,当我们有多台电脑,多台CPU的时候我们只需要把这个大的岛分成几个小的岛,然后每一台CPU各自运行,这样的效率肯定快很多。
这里介绍一种单CPU,单内存解决这个问题的方法。
依次遍历每个元素,当它为1时,把它的上下左右都给感染了变成2,当元素不为1的时候,就跳过这个元素,进行下一个。为1的时候岛的数量就加一。
代码:
public class Code_03_Islands {
public static int countIslands(int[][] m) {
if (m == null || m[0] == null) {
return 0;
}
int N = m.length; //行长
int M = m[0].length;//列长
int res = 0;
for (int i = 0; i < N; i++) {
for (int j = 0; j < M; j++) {
if (m[i][j] == 1) {
res++; //岛的数量加1
infect(m, i, j, N, M);
}
}
}
return res;
}
public static void infect(int[][] m, int i, int j, int N, int M) {
if (i < 0 || i >= N || j < 0 || j >= M || m[i][j] != 1) { //如果这里不是1的话,就返回
return;
}
m[i][j] = 2; //把当前元素变成2
infect(m, i + 1, j, N, M); //依次去感染它的上下左右
infect(m, i - 1, j, N, M);
infect(m, i, j + 1, N, M);
infect(m, i, j - 1, N, M);
}
public static void main(String[] args) {
int[][] m1 = { { 0, 0, 0, 0, 0, 0, 0, 0, 0 },
{ 0, 1, 1, 1, 0, 1, 1, 1, 0 },
{ 0, 1, 1, 1, 0, 0, 0, 1, 0 },
{ 0, 1, 1, 0, 0, 0, 0, 0, 0 },
{ 0, 0, 0, 0, 0, 1, 1, 0, 0 },
{ 0, 0, 0, 0, 1, 1, 1, 0, 0 },
{ 0, 0, 0, 0, 0, 0, 0, 0, 0 }, };
System.out.println(countIslands(m1));
int[][] m2 = { { 0, 0, 0, 0, 0, 0, 0, 0, 0 },
{ 0, 1, 1, 1, 1, 1, 1, 1, 0 },
{ 0, 1, 1, 1, 0, 0, 0, 1, 0 },
{ 0, 1, 1, 0, 0, 0, 1, 1, 0 },
{ 0, 0, 0, 0, 0, 1, 1, 0, 0 },
{ 0, 0, 0, 0, 1, 1, 1, 0, 0 },
{ 0, 0, 0, 0, 0, 0, 0, 0, 0 }, };
System.out.println(countIslands(m2));
}
}