Leetcode-419. 甲板上的战舰

链接

力扣https://leetcode-cn.com/problems/battleships-in-a-board/

题目

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

思路一 DFS

遍历图上所有符号位'X'的点，找到一个点遍进行计数+1，然后对该点进行深度优先搜索，将搜到的同样为‘X’的格子标记为‘.’，防止后面遍历时重复计数（因为连在一起的X为同一艘战舰），最后返回计数即可。

C++ Code

class Solution {
public:
    void dfs(int i,int j,vector<vector<char>>& board)
    {
        if(i<0||i>=board.size()||j<0||j>=board[0].size()||board[i][j]=='.') return;
        board[i][j]='.';
        dfs(i-1,j,board);
        dfs(i+1,j,board);
        dfs(i,j+1,board);
        dfs(i,j-1,board);
    }

    int countBattleships(vector<vector<char>>& board) {
        int count=0;
        for(int i=0;i<board.size();i++)
        {
            for(int j=0;j<board[0].size();j++)
            if(board[i][j]=='X')
            {
                count++;
                dfs(i,j,board);
            }
        }
        return count;

    }
};

结果

 思路二 原地判断

每艘战舰要么是竖的，要么是横的，且两艘战舰之间至少有一个空格，因此每艘战舰的起点的上方和左方都不会是 'X'，所以，我们只需要统计起点即可。

C++ Code

class Solution {
public:
    int countBattleships(vector<vector<char>>& board) {
        int cnt = 0, m = board.size(), n = board[0].size();
        for(int i = 0; i < m; ++i)
            for(int j = 0; j < n; ++j)
                // 起点为X，左边不为X，上边不为X
                if(board[i][j] == 'X' && (!i || board[i-1][j] != 'X') && (!j || board[i][j-1] != 'X')) 
                    ++cnt;
        return cnt;
    }
};

结果