LeetCode-Python-419. 甲板上的战舰

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

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

示例 :

X..X
...X
...X

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

无效样例 :

...X
XXXX
...X

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

进阶:

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

基础思路：

DFS搜索+染色。

找到一个X就在他附近找战舰的其他部分，然后把用过的染色。

class Solution(object):
    def countBattleships(self, board):
        """
        :type board: List[List[str]]
        :rtype: int
        """
        
        m = len(board)
        if m == 0:
            return 0
        n = len(board[0])
        if n == 0:
            return 0
        res = 0    
        def dfs(x, y):
            dx = [1, -1, 0, 0]
            dy = [0, 0, 1, -1]
            
            for k in range(4):
                xx = x + dx[k]
                yy = y + dy[k]
                
                if 0 <= xx < m and 0 <= yy < n and board[xx][yy] == "X":
                    board[xx][yy] = "."
                    dfs(xx, yy)
                    
        for i in range(m):
            for j in range(n):
                if board[i][j] == "X":
                    # print 1
                    board[i][j] = "."
                    dfs(i, j)
                    res += 1
                    
        return res