望着时间滴答滴答的流过,我不曾改变过 . . .

419. Battleships in a Board

题目描述:
用一次遍历,找出来有多少个不相连的X
第一遍,自己写的

def countBattleships(self, board: List[List[str]]) -> int:
        res = 0
        for i in range(len(board)):
            for j in range(len(board[0])):
                if board[i][j] == 'X':
                    res += 1
                    board[i][j] = 'Y'
                    for m in range(i + 1,len(board)):
                        if board[m][j] == 'X':
                            board[m][j] = 'Y'
                        else:
                            break
                            
                    for n in range(j + 1,len(board[0])):
                        if board[i][n] == 'X':
                            board[i][n] = 'Y'
                        else:
                            break
                elif board[i][j] == 'Y':
                    board[i][j] == 'X'
        
        return res

看完答案,果然自己又弟弟了
只要遍历的时候,计数:计最左上角的就可以

关键在于:最左上角的条件怎么去写?
观察可以得出:最左上角的元素或者是 上方是 ‘.’ 或者是左方是'.' (因为不是最左上角的元素,上方或者左方必定是x)

同时,如果最左上角的元素, index == 0 也要考虑进来。
代码如下:

def countBattleships(self, board: List[List[str]]) -> int:
        res = 0
        for i in range(len(board)):
            for j in range(len(board[0])):
                if board[i][j] == 'X' and (i == 0 or board[i-1][j] == '.') and (j == 0 or board[i][j-1] == '.'):
                    res += 1
        
        return res
posted @ 2019-05-13 14:15  whyaza  阅读(125)  评论(0编辑  收藏  举报