python 战舰_419. 甲板上的战舰（Python）

题目

难度：★★★☆☆

类型：二维数组

方法：深度优先搜索

力扣链接请移步本题传送门

更多力扣中等题的解决方案请移步力扣中等题目录

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

给你一个有效的甲板，仅由战舰或者空位组成。

战舰只能水平或者垂直放置。换句话说,战舰只能由 1xN (1 行, N 列)组成，或者 Nx1 (N 行, 1 列)组成，其中N可以是任意大小。

两艘战舰之间至少有一个水平或垂直的空位分隔 - 即没有相邻的战舰。

示例 :

X..X

...X

...X

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

无效样例 :

...X

XXXX

...X

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

进阶:

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

解答

这道题与统计岛屿数目是一样的。

解法1，我们可以采用深度优先搜索算法来实现，遍历每个元素，如果遇到战舰，则用深度优先搜索将战舰抹去，同时计数器+1。

class Point:

"""

点的实例，包含获取左边、右边、上边、下边相邻点和获取当前坐标值的方法。

"""

def __init__(self, h, w):

self.h = h

self.w = w

@property

def left(self):

return Point(self.h, self.w-1)

@property

def right(self):

return Point(self.h, self.w+1)

@property

def upper(self):

return Point(self.h-1, self.w)

@property

def lower(self):

return Point(self.h+1, self.w)

@property

def locate(self):

return [self.h, self.w]

class Board:

"""

网格实例，用于构建网格，包含取某点值，设置某点值，判断点是否在网格内等方法。

"""

def __init__(self, value):

value = value if value else [[]]

self.value = value

self.height = len(value)

self.width = len(value[0])

def get(self, point):

assert isinstance(point, Point)

if self.is_valid(point):

return self.value[point.h][point.w]

return -1

def set(self, point, value):

assert isinstance(point, Point)

if self.is_valid(point):

self.value[point.h][point.w] = value

return True

return -1

def is_valid(self, point):

assert isinstance(point, Point)

return 0 <= point.w < self.width and 0 <= point.h < self.height

class Deck:

def __init__(self, value):

self.deck = Board(value)

self.width = self.deck.width

self.height = self.deck.height

@property

def warship(self):

s = 0

def erase(point):

if self.deck.get(point) == 'X':

self.deck.set(point, '.')

erase(point.left)

erase(point.right)

erase(point.upper)

erase(point.lower)

for i in range(self.height):

for j in range(self.width):

if self.deck.get(Point(i, j)) == 'X':

s += 1

erase(Point(i, j))

return s

class Solution:

def countBattleships(self, board):

deck = Deck(board)

return deck.warship

if __name__ == "__main__":

s = Solution()

data = [['X', '.', '.', 'X',],

['.', '.', '.', 'X',],

['.', '.', '.', 'X',],

]

print(s.countBattleships(data))

解法2：我们也可以统计所有左上相邻元素均为“.”的“X”的个数，就是战舰的个数。

class Solution:

def countBattleships(self, board):

height, width = len(board), len(board[0])

s = 0

for h in range(height):

for w in range(width):

if board[h][w] == 'X' and (h == 0 or board[h-1][w] == '.') and (w == 0 or board[h][w-1] == '.'):

s += 1

return s

精简版一行代码实现：

class Solution:

def countBattleships(self, board):

return sum([board[h][w] == 'X' and (h == 0 or board[h-1][w] == '.') and (w == 0 or board[h][w-1] == '.') for w in range(len(board[0])) for h in range(len(board))])

如有疑问或建议，欢迎评论区留言~

有关更多力扣中等题的python解决方案，请移步力扣中等题解析