牛客网 BM57.
二维矩阵,值为'1'表示岛屿,'0'表示海洋,求海洋中岛屿数量。
解题思路:
遍历二维数组,值为1增加岛屿数量记数,同时对此位置进行单独递归遍历上下左右4个方向,将数组坐标范围内同时值为1的元素置0,同时递归进入下一层,直到将所有能遍历到的1全部置0,处理完再返回主程序遍历二维数组下一个位置,直到遍历完成。
代码:
// gopackage main
// import "fmt"/*** 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可** 判断岛屿数量* @param grid char字符型二维数组 * @return int整型
*/func process(grid [][]byte, i, j, m, n int) {grid[i][j] = '0'// 上if i - 1 >= 0 && grid[i-1][j] == '1'{process(grid, i-1, j, m, n)}// 下if i + 1 < m && grid[i+1][j] == '1' {process(grid, i+1, j, m, n)}// 左if j - 1 >= 0 && grid[i][j-1] == '1' {process(grid, i, j-1, m, n)}// 右if j + 1 < n&& grid[i][j+1] == '1' {process(grid, i, j+1, m, n)}
}func solve( grid [][]byte ) int {// write code hereif len(grid) == 0 || len(grid[0]) == 0 {return 0}var res = 0m := len(grid)n := len(grid[0])for i := 0; i < m; i++ {for j := 0; j < n; j++ {if grid[i][j] == '1' {res++process(grid, i, j, m, n)}}}return res
}