牛客网 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
}