给定一个二维的甲板, 请计算其中有多少艘战舰。 战舰用 'X’表示,空位用 '.'表示。 你需要遵守以下规则:
给你一个有效的甲板,仅由战舰或者空位组成。
战舰只能水平或者垂直放置。换句话说,战舰只能由 1xN
(1 行, N 列)组成,或者 Nx1
(N 行, 1 列)组成,其中N可以是任意大小。
两艘战舰之间至少有一个水平或垂直的空位分隔 - 即没有相邻的战舰。
-
示例 :
X…X
…X
…X
在上面的甲板中有2艘战舰。 -
无效样例 :
…X
XXXX
…X
你不会收到这样的无效甲板 - 因为战舰之间至少会有一个空位将它们分开。 -
进阶:
你可以用一次扫描算法,只使用O(1)额外空间,并且不修改甲板的值来解决这个问题吗?
class Solution:
def countBattleships(self, board: List[List[str]]) -> int:
#只需要一次遍历 只要该位置为‘X’并且左边和上边都是‘.’,那这个位置就是战舰的头部,只需要数战舰的头部就好了
count = 0
for i in range(len(board)): #行
for j in range(len(board[0])): #列
#通常用or短路后面的语句防止索引越界
if (board[i][j]=='X') and (i==0 or board[i-1][j]=='.') and (j==0 or board[i][j-1]=='.'):
count+=1
return count
#我们可以通过战舰的头来判断个数,当一个点上面或者左面试X说明它战舰中间部分,跳过即可!
# row = len(board)
# col = len(board[0])
# res = 0
# for i in range(row):
# for j in range(col):
# 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
二维数组遍历常用代码:
row = len(board)
col = len(board[0])
for i in range(row):
for j in range(col):
board[i][j]