要求
给定一个二维的甲板, 请计算其中有多少艘战舰。 战舰用 '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
解题思路:就是二层循环遍历,因为我们是从第一行中向下不断地搜索,所以遇到空位或者上一行中有飞机、或者旁边列中有飞机,都不是有效的飞机,比较简单,重在理解题意。