牛客网 BM57.
二维矩阵,值为'1'表示岛屿,'0'表示海洋,求海洋中岛屿数量。
解题思路:
遍历二维数组,值为1增加岛屿数量记数,同时对此位置进行单独递归遍历上下左右4个方向,将数组坐标范围内同时值为1的元素置0,同时递归进入下一层,直到将所有能遍历到的1全部置0,处理完再返回主程序遍历二维数组下一个位置,直到遍历完成。
代码:
- // go
-
- package 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 here
- if len(grid) == 0 || len(grid[0]) == 0 {
- return 0
- }
- var res = 0
- m := 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
- }