python解數獨

昨晚心血來潮在leetcode上pick one了一道算法題

https://leetcode.com/problems/sudoku-solver/

解決代碼如下：

class Solution(object):
    def solveSudoku(self, board):
        """
        :type board: List[List[str]]
        :rtype: void Do not return anything, modify board in-place instead.
        """
        def check(i, j, value):
            for item in board[i]:
                if item == value:
                    return False
            for item in board:
                if item[j] == value:
                    return False
            col, row = i/3*3, j/3*3
            array = board[col][row:row+3] + board[col+1][row:row+3] + board[col+2][row:row+3]
            for item in array:
                if item == value:
                    return False
            return True
        def get_next(i, j): 
            for row in range(j+1, 9): 
                if board[i][row] == ".":
                    return i, row 
            for col in range(i+1, 9): 
                for row in range(9):
                    if board[col][row] == ".":
                        return col, row 
            return -1, -1
        def try_num(i, j): 
            if board[i][j] == ".":
                for num in range(1, 10):
                    if check(i, j, str(num)):
                        board[i][j] = str(num)
                        next_i, next_j = get_next(i, j)
                        if next_i == -1: 
                            return True
                        else:
                            end = try_num(next_i, next_j)
                            if not end:
                                board[i][j] = "."
                            else:
                                return True
        if board[0][0] == ".":
            try_num(0, 0)
        else:
            i, j = get_next(0, 0)
            try_num(i, j)
        return

主要使用回溯遞歸的方法，先定義一個判斷函數和一個獲得下一個位置的函數，使結構清晰一些。

然后對可選i,j進行1~9遍歷，如果遍歷成功，獲得next_i, next_j進行下一輪遍歷，如果失敗則重置為"."

思路還是挺好理解的，代碼beat了60%，不知道為什么這個題設為hard了，以前設計過數獨的題，這次解數獨題也不是什么難事了。

 

突然想起來test case給的是

["..9748...","7........",".2.1.9...","..7...24.",".64.1.59.",".98...3..","...8.3.2.","........6","...2759.."]

實際給的是

[['.', '.', '9', '7', '4', '8', '.', '.', '.'], ['7', '.', '.', '.', '.', '.', '.', '.', '.'], ['.', '2', '.', '1', '.', '9', '.', '.', '.'], ['.', '.', '7', '.', '.', '.', '2', '4', '.'], ['.', '6', '4', '.', '1', '.', '5', '9', '.'], ['.', '9', '8', '.', '.', '.', '3', '.', '.'], ['.', '.', '.', '8', '.', '3', '.', '2', '.'], ['.', '.', '.', '.', '.', '.', '.', '.', '6'], ['.', '.', '.', '2', '7', '5', '9', '.', '.']]

自己測試的時候別弄混了