题目
给定一个二维的甲板, 请计算其中有多少艘战舰。 战舰用 'X’表示,空位用 '.'表示。 你需要遵守以下规则:
给你一个有效的甲板,仅由战舰或者空位组成。
战舰只能水平或者垂直放置。换句话说,战舰只能由 1xN (1 行, N 列)组成,或者 Nx1 (N 行, 1 列)组成,其中N可以是任意大小。
两艘战舰之间至少有一个水平或垂直的空位分隔 - 即没有相邻的战舰。
示例 :
X…X
…X
…X
在上面的甲板中有2艘战舰。
无效样例 :
…X
XXXX
…X
你不会收到这样的无效甲板 - 因为战舰之间至少会有一个空位将它们分开。
进阶:
你可以用一次扫描算法,只使用O(1)额外空间,并且不修改甲板的值来解决这个问题吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/battleships-in-a-board
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思想
1.原题思想可以参考第200题,岛屿数量;(https://blog.csdn.net/wszhang_/article/details/107635138)
2.进阶:当我们遇到X,如果它的上一行对应列也是X,那么这个X应该处于竖着的战舰的中间位置;同理,如果它的左边也是X,那就应该处于横着战舰的中间位置。这两者情况都是可以战舰数量加1的。如果X是出于战舰的拐角,如它的上面是没有的,左边是没有的,这种情况也是可以战舰数量加1的。
代码
// atom-one-night
var countBattleships = function(board) {
let count = 0;
for(let row=0;row<board.length;row++){
for(let col=0;col<board[0].length;col++){
if(board[row][col] === "X"){
count++;
dfs(row, col)
}
}
}
function dfs(row, col){
if(row<0 || row>= board.length || col<0 || col>=board[0].length || board[row][col] !== "X"){
return ;
}
board[row][col] = ".";
dfs(row-1, col);
dfs(row+1, col);
dfs(row,col-1);
dfs(row,col+1);
}
return count;
}
进阶:
// atom-one-night
var countBattleships = function(board) {
let count = 0;
for(let row=0;row<board.length;row++){
for(let col=0;col<board[0].length;col++){
if(board[row][col] === "X"){
if(row>0 && board[row-1][col] === "X"){
continue;
}
if(col>0 && board[row][col-1] === "X"){
continue;
}
count++;
}
}
}
}