问题描述
给你一个大小为
m x n
的矩阵board
表示甲板,其中,每个单元格可以是一艘战舰'X'
或者是一个空位'.'
,返回在甲板board
上放置的 战舰 的数量。战舰 只能水平或者垂直放置在
board
上。换句话说,战舰只能按1 x k
(1
行,k
列)或k x 1
(k
行,1
列)的形状建造,其中k
可以是任意大小。两艘战舰之间至少有一个水平或垂直的空位分隔 (即没有相邻的战舰)。
示例
示例 1:
输入:board = [["X",".",".","X"],[".",".",".","X"],[".",".",".","X"]] 输出:2示例 2:
输入:board = [["."]] 输出:0提示:
m == board.length
n == board[i].length
1 <= m, n <= 200
board[i][j]
是'.'
或'X'
问题分析:
今天的每日一题,进阶要求是O(1)的空间复杂度,并且不修改board的值,现在的我完全写不出来。。。。。。说一下我的做法,类似岛屿问题,直接dfs。增加辅助数组visited来记录该战舰是否已经被计算到。
代码如下:
class Solution {
public:
// 战舰只有两个方向,→或者↓
int dir[2][2] = {{0, 1}, {1, 0}};
int countBattleships(vector<vector<char>>& board) {
int ans = 0;
int n = board.size();
int m = board[0].size();
vector<vector<bool>> visited(n, vector<bool>(m, false));
// 炫耀一下新学会的lambda表达式hhh
function<void(int, int)> dfs = [&](int i, int j){
for(int d = 0; d < 2; ++d){
int ti = i + dir[d][0];
int tj = j + dir[d][1];
if(ti >= n || tj >= m || board[ti][tj] != 'X' || visited[ti][tj]) continue;
visited[ti][tj] = true;
dfs(ti, tj);
}
};
// 遍历遇到一个X就计数加一并使用dfs函数标记该战舰其余部分
for(int i = 0; i < n; ++i){
for(int j = 0; j < m; ++j){
if(visited[i][j] == false && board[i][j] == 'X'){
++ans;
visited[i][j] = true;
dfs(i, j);
}
}
}
return ans;
}
};