leetcode刷题记录35-419. 甲板上的战舰

问题描述

给你一个大小为 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;
    }
};