LeetCode题解 | [中等-数组] 419. 甲板上的战舰

419. 甲板上的战舰

题目

给定一个二维的甲板， 请计算其中有多少艘战舰。 战舰用 'X'表示，空位用 '.'表示。 你需要遵守以下规则：

给你一个有效的甲板，仅由战舰或者空位组成。
战舰只能水平或者垂直放置。换句话说,战舰只能由 1xN (1 行, N 列)组成，或者 Nx1 (N 行, 1 列)组成，其中N可以是任意大小。
两艘战舰之间至少有一个水平或垂直的空位分隔 - 即没有相邻的战舰。

示例 :

X..X
...X
...X
在上面的甲板中有2艘战舰。

无效样例 :

...X
XXXX
...X
你不会收到这样的无效甲板 - 因为战舰之间至少会有一个空位将它们分开。

进阶:
你可以用一次扫描算法，只使用O(1)额外空间，并且不修改甲板的值来解决这个问题吗？

思路

1. 简单暴力扫描+扩展
2. 模式判断法，通过判断该点是否为战舰的左/上头部来标定一艘战舰

代码

/*
2021/2/2
暴力算法
*/
class Solution {
public:
    int countBattleships(vector<vector<char>>& board) {
        const int len = board.size();
        const int width = board[0].size();

        int i, j, k;
        int cnt = 0;

        for (i = 0; i < len; i++) {
            for (j = 0; j < width; j++) {
                if (board[i][j] == 'X') {
                    ++cnt;
                    board[i][j] = '.';
                    //横向扩展
                    for (int k = j+1; k < width && board[i][k] == 'X'; k++) {
                        board[i][k] = '.';
                    }
                    //纵向扩展
                    for (int k = i+1; k < len && board[k][j] == 'X'; k++) {
                        board[k][j] = '.';
                    }
                }
            }
        }

        return cnt;
    }
};

//模式判断
class Solution {
public:
    int countBattleships(vector<vector<char>>& board) {
        const int len = board.size();
        const int width = board[0].size();

        int i, j, k;
        int cnt = 0;

        for (i = 0; i < len; i++) {
            for (j = 0; j < width; j++) {
                if (board[i][j] == 'X') {
                    //单点
                    if ((i-1 < 0 || board[i-1][j] == '.') && 
                        (i+1 >= len || board[i+1][j] == '.') &&
                        (j-1 < 0 || board[i][j-1] == '.') &&
                        (j+1 >= width || board[i][j+1] == '.')) {
                        
                        ++cnt;
                    }
                    //横向
                    if ((i-1 < 0 || board[i-1][j] == '.') && 
                        (i+1 >= len || board[i+1][j] == '.') &&
                        (j-1 < 0 || board[i][j-1] == '.') &&
                        (j+1 < width && board[i][j+1] == 'X')) {
                        
                        ++cnt;
                    }
                    //纵向
                    if ((i-1 < 0 || board[i-1][j] == '.') && 
                        (i+1 < len && board[i+1][j] == 'X') &&
                        (j-1 < 0 || board[i][j-1] == '.') &&
                        (j+1 >= width || board[i][j+1] == '.')) {
                        
                        ++cnt;
                    }
                }
            }
        }

        return cnt;
    }
};