一、每日一题
题意
这题的标题应该是《棋盘上的战舰》,来源于 海战棋,把横着或竖着的连续 X 看成一艘战舰,统计棋盘上有多少艘战舰。
思路
战舰的个数,等于战舰「头部」的个数。如下图,我们只需要统计蓝色 X 的个数,即为战舰的个数。
具体来说,如果位于 (i,j)(i,j)(i,j) 的格子是战舰的头部,那么左边和上边的相邻格子不能是 X,即:
如果 j>0,那么 (i,j−1)(i,j-1)(i,j−1) 不能是 X。
如果 i>0,那么 (i−1,j)(i-1,j)(i−1,j) 不能是 X。
解答:
def countBattleships(board):
if not board:
return 0
m, n = len(board), len(board[0])
count = 0
for i in range(m):
for j in range(n):
if board[i][j] == 'X':
# 检查'X'是否是新船
if (i == 0 or board[i-1][j] == '.') and (j == 0 or board[i][j-1] == '.'):
count = count + 1
return count
# 示例
board1 = [["X",".",".","X"],[".",".",".","X"],[".",".",".","X"]]
print(countBattleships(board1)) # 输出: 2
board2 = [["."]]
print(countBattleships(board2)) # 输出: 0
题源:Leetcode
二、总结
理解题意,通过枚举,关键是想到if的条件设置。
2024.6.11