题目
给你一个大小为 m x n 的矩阵 board 表示甲板,其中,每个单元格可以是一艘战舰 ‘X’ 或者是一个空位 ‘.’ ,返回在甲板 board 上放置的 战舰 的数量。 战舰 只能水平或者垂直放置在 board 上。换句话说,战舰只能按 1 x k(1 行,k 列)或 k x 1(k 行,1 列)的形状建造,其中 k 可以是任意大小。两艘战舰之间至少有一个水平或垂直的空位分隔 (即没有相邻的战舰)。
思路
- 就是数有几个船, 就是数就完了。
代码
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