leetcode 200. 岛屿数量 golang实现

爱⌒轻易说出口 提交于 2020-01-15 01:05:40
描述
给定一个由 '1'(陆地)和 '0'(水)组成的的二维网格,计算岛屿的数量。一个岛被水包围,并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。

示例 1:

输入:
11110
11010
11000
00000

输出: 1
示例 2:

输入:
11000
11000
00100
00011

输出: 3
思路
深度优先遍历
每次遇到=1的地方 岛屿数+1 上下左右开始遍历 将遇到的1的位置全部置为0 遇到0的地方或者遍历返回
实现
func numIslands(grid [][]byte) int {
    var count int
    if len(grid) == 0{
        return count
    }
    rows := len(grid)
    cols := len(grid[0])

    for i := 0; i < rows; i++{
        for j := 0; j < cols; j++{
            if grid[i][j] == '1'{
                dfs(grid, i, j, rows, cols)
                count++
            }
        }
    }

    return count
}

func dfs(grid [][]byte, i, j, rows, cols int){
    if i < 0 || j < 0 || i >= rows || j >= cols || grid[i][j] == '0'{
        return
    }

    grid[i][j] = '0'
    dfs(grid, i-1, j, rows, cols)
    dfs(grid, i+1, j, rows, cols)
    dfs(grid, i, j-1, rows, cols)
    dfs(grid, i, j+1, rows, cols)

    return
}
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!