编程语言
首页 > 编程语言> > 深度优先搜索简单算法题

深度优先搜索简单算法题

作者:互联网

给你一个大小为 m x n 的矩阵 board 表示甲板,其中,每个单元格可以是一艘战舰 'X' 或者是一个空位 '.' ,返回能在甲板 board 上放置的 战舰 的数量。

战舰 只能水平或者垂直放置在 board 上。换句话说,战舰只能按 1 x k(1 行,k 列)或 k x 1(k 行,1 列)的形状建造,其中 k 可以是任意大小。两艘战舰之间至少有一个水平或垂直的空位分隔 (即没有相邻的战舰)。

输入:board = [["X",".",".","X"],[".",".",".","X"],[".",".",".","X"]]
输出:2

我们这道题采用深度优先遍历方法搜索于"X"相邻的"X",并把他们替换成"."

详细代码如下

var countBattleships = function(board) {

    var s = 0;

    for(let x = 0;x<board.length;x++){

        for(let y = 0;y<board[0].length;y++){

            if(board[x][y]==='X'){

                s++;

                dfs(x,y)

            }

        }

    }

    function dfs(x,y){

        if(x<0||x>=board.length||y<0||y>=board[0].length||board[x][y]=='.'){

            return

        }

        board[x][y]='.'

        dfs(x+1,y)

        dfs(x-1,y)

        dfs(x,y+1)

        dfs(x,y-1)

    }

    return s

};

dfs中采用的就是递归的思想来搜索与“X”相邻的“X”。

标签:function,优先,return,dfs,算法,搜索,board,var,战舰
来源: https://blog.csdn.net/qq_52819902/article/details/121581865