419 甲板上的战舰

本文最后更新于:2022年4月9日 中午


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

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

示例 :

1
2
3
X..X
...X
...X

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

无效样例 :

1
2
3
...X
XXXX
...X

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

进阶:

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

Solution

由于同一艘战舰的X都是相连的,对于坐标为 [i, j] 值如果为X,up_pos=[i-1,j],left_pos=[i, j-1]

  • up_pos 和 left_pos 都不是X,判断 [i, j] 为一个战舰的起始。
  • up_pos 或 left_pos 如果为X,说明是同一艘战舰。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# @lc code=start
class Solution:
def countBattleships(self, board: List[List[str]]) -> int:
row = len(board)
col = len(board[0])
cnt = 0
for i in range(row):
for j in range(col):
up_pos = False
left_pos = False
if board[i][j] == 'X':
if i-1 >= 0:
if board[i-1][j] == 'X':
up_pos = True
if j-1 >= 0:
if board[i][j-1] == 'X':
left_pos = True
if not (up_pos or left_pos):
cnt += 1
return cnt
# @lc code=end

本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!