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 的值来解决这个问题吗？

思路

做过1992题过来的，解法相同，不过这个题因为战舰都是长条形的，所以代码更简洁一些。不好想的是如何不修改输入保持空间#card=math&code=O%281%29&id=NuyXi)

代码

修改输入

class Solution {
    public int countBattleships(char[][] board) {
        int n = board.length;
        int m = board[0].length;
        int ans = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (board[i][j] == '.') {
                    continue;
                }
                board[i][j] = '.';
                // 两个while循环应该分别从i+1和j+1开始，不然可能第一个修改了board之后无法进入下面的while
                int r = i + 1;
                while (r < n && board[r][j] == 'X') {
                    board[r][j] = '.';
                    r++;
                }
                int c = j + 1;
                while (c < m && board[i][c] == 'X') {
                    board[i][c] = '.';
                    c++;
                }
                ans++;
            }
        }
        return ans;
    }
}

不修改输入

思路来自官解法二：因为题目中给定的两艘战舰之间至少有一个水平或垂直的空位分隔，任意两个战舰之间是不相邻的，因此我们可以通过枚举每个战舰的左上顶点即可统计战舰的个数。

如果一个位置是战舰，左面和上面都不是战舰，那么就找到了一个战舰的左上角。

class Solution {
    public int countBattleships(char[][] board) {
        int n = board.length;
        int m = board[0].length;
        int ans = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (board[i][j] == 'X') {
                    if ((i == 0 || board[i - 1][j] != 'X') && (j == 0 || board[i][j - 1] != 'X')) {
                        ans++;
                    }
                }
            }
        }
        return ans;
    }
}