给你一个大小为 m x n 的矩阵 board 表示甲板,其中,每个单元格可以是一艘战舰 ‘X’ 或者是一个空位 ‘.’ ,返回在甲板
board 上放置的 战舰 的数量。战舰 只能水平或者垂直放置在 board 上。换句话说,战舰只能按 1 x k(1 行,k 列)或 k x 1(k 行,1
列)的形状建造,其中 k 可以是任意大小。两艘战舰之间至少有一个水平或垂直的空位分隔 (即没有相邻的战舰)。
输入:board = [
[“X”,“.”,“.”,“X”],
[“.”,“.”,“.”,“X”],
[“.”,“.”,“.”,“X”]]
输出:2
示例 2:
输入:board = [[“.”]]
输出:0
解题思路:
1、直接用朴素的"感染"思想即可
2、即任意两艘战舰是不相邻的,所以每次遇到一艘战舰,我们先计数,再感染掉它,避免了重复计算
代码:
class Solution {
public int res = 0;
public int countBattleships(char[][] board) {
for(int i = 0; i < board.length; i ++)
for(int j = 0; j < board[0].length; j ++)
if(board[i][j] == 'X') {
dfs(i, j, board);
res ++;
}
return res;
}
public void dfs(int x, int y, char board[][]) {
board[x][y] = 'z';//感染
int fxx = 0;
int fyy = 0;
int fx[] = {-1, 1, 0, 0};
int fy[] = {0, 0, -1, 1};
for(int i = 0; i < 4; i ++) {
fxx = fx[i] + x;
fyy = fy[i] + y;
if(fxx >= 0 && fxx < board.length && fyy >= 0 && fyy < board[0].length && board[fxx][fyy] == 'X') dfs(fxx, fyy, board);
}
}
}