[LeetCode] 419. 甲板上的战舰

题目

给你一个大小为 m x n 的矩阵 board 表示甲板，其中，每个单元格可以是一艘战舰 ‘X’ 或者是一个空位 ‘.’ ，返回在甲板 board 上放置的 战舰 的数量。 战舰 只能水平或者垂直放置在 board 上。换句话说，战舰只能按 1 x k（1 行，k 列）或 k x 1（k 行，1 列）的形状建造，其中 k 可以是任意大小。两艘战舰之间至少有一个水平或垂直的空位分隔 （即没有相邻的战舰）。

思路

1. 就是数有几个船， 就是数就完了。

代码

class Solution:
    def countBattleships(self, board: List[List[str]]) -> int:
        ret, n, m = 0, len(board), len(board[0])
        def p(i, j):
            return True if not (0 <= i < n and 0 <= j < m) else board[i][j] == '.'
        for i in range(n):
            for j in range(m):
                if board[i][j] == 'X' and p(i-1, j) and p(i, j-1):
                    ret += 1
        return ret