1、题目描述
有一个由0和1组成的二维矩阵,其中1代表陆地,0代表水,岛屿由水平或垂直方向上相邻的陆地连接形成。
假设矩阵的四周均被水包围,请计算岛屿的数量。
输入:matrix =[[1,1,0,0],[0,0,1,0],[0,0,0,0],[0,0,1,1],],
如图所示:
—————————————————
| 1 | 1 | | |
—————————————————
| | | 1 | |
—————————————————
| | | | |
—————————————————
| | | 1 | 1 |
—————————————————
输出:3
2、 题目解析
扫描整个矩阵,每遇到1就将它置为0,并以它为起始节点开始进行深度优先搜索。最终岛屿的数量就是进行深度优先搜索的次数。
3、代码实现
public class NumOfLands {private static int ans = 0;public static void main(String[] args) {int[][] matrix = {{1, 1, 0, 0}, {0, 0, 1, 0}, {0, 0, 0, 0}, {0 ,0, 1, 1}};System.out.println(numOfLands(matrix));}public static int numOfLands(int[][] matrix) {if (matrix == null) {return 0;}// 岛屿的数量就是进行深度优先搜索的次数for (int i = 0; i < matrix.length; i++) {for (int j = 0; j < matrix[0].length; j++) {if (matrix[i][j] == 1) {ans ++;dfs(matrix, i, j);}}}return ans;}private static void dfs(int[][] matrix, int i, int j) {// 搜索的边界if (i < 0 || i >= matrix.length || j < 0 || j >= matrix[0].length) {return;}// 每次遇到1就将它置为0,并以它为起始节点开始进行深度优先搜索if (matrix[i][j] == 1) {matrix[i][j] = 0;// 搜索四个方向:上、下、左、右dfs(matrix, i - 1, j);dfs(matrix, i + 1, j);dfs(matrix, i, j - 1);dfs(matrix, i, j + 1);}}}