leetcode_419 甲板上的战舰

99 阅读1分钟

要求

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

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

无效样例 :

...X
XXXX
...X

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

核心代码

class Solution:
    def countBattleships(self, board: List[List[str]]) -> int:
        res = 0
        m,n = len(board),len(board[0])
        for i in range(m):
            for j in range(n):
                if board[i][j] == ".":
                    continue
                if i > 0 and board[i - 1][j] == "X":
                    continue
                if j > 0 and board[i][j - 1] == "X":
                    continue
                res += 1
        return res

image.png

解题思路:就是二层循环遍历,因为我们是从第一行中向下不断地搜索,所以遇到空位或者上一行中有飞机、或者旁边列中有飞机,都不是有效的飞机,比较简单,重在理解题意。