算法 岛屿数量-(递归回溯)

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

你可能感兴趣的:(牛客网,算法刷题,岛屿数量,递归回溯)