代码随想录算法训练营第三十天 | leetcode 332.重新安排行程，51. N皇后，37. 解数独

332.重新安排行程

题目：给定一个机票的字符串二维数组 [from, to]，子数组中的两个成员分别表示飞机出发和降落的机场地点，对该行程进行重新规划排序。所有这些机票都属于一个从 JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 开始。

题目链接：332. 重新安排行程

class Solution:
    def findItinerary(self, tickets: List[List[str]]) -> List[str]:
        # defaultdic(list) 是为了方便直接append
        tickets_dict = defaultdict(list)
        for item in tickets:
            tickets_dict[item[0]].append(item[1])
        # 给每一个机场的到达机场排序，小的在前面，在回溯里首先被pop(0）出去
	    # 这样最先找的的path就是排序最小的答案，直接返回
        for airport in tickets_dict:
            tickets_dict[airport].sort()

        path = ["JFK"]
        def backtracking(start_point):
            if len(path) == len(tickets) + 1:
                return True
            for _ in tickets_dict[start_point]:
                # 必须及时删除，避免出现死循环
                end_point = tickets_dict[start_point].pop(0)
                path.append(end_point)
                # 只要找到一个就可以返回了
                if backtracking(end_point):
                    return True
                path.pop()
                tickets_dict[start_point].append(end_point)
        backtracking("JFK")
        return path

51. N皇后

题目：n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。
给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

题目链接：51. N 皇后

class Solution:
    def solveNQueens(self, n: int) -> List[List[str]]:
        if not n:
            return []
        board = [['.'] * n for _ in range(n)]
        res = []
        def isValid(board, row, col):
            for i in range(len(board)): # 判断同一列是否冲突
                if board[i][col] == 'Q':
                    return False
            # 判断左上角是否冲突
            i = row - 1  
            j = col - 1
            while i >= 0 and j >= 0:
                if board[i][j] == 'Q':
                    return False
                j -= 1
                i -= 1
            # 判断右上角是否冲突
            i = row - 1
            j = col + 1
            while i >= 0 and j < len(board):
                if board[i][j] == 'Q':
                    return False
                i -= 1
                j += 1
            return True

        def backtracking(board, row, n):
            if row == n:  # 如果走到最后一行，说明已经找到一个解
                temp_res = []
                for temp in board:
                    temp_str = "".join(temp)
                    temp_res.append(temp_str)
                res.append(temp_res)
            for col in range(n):
                if not isValid(board, row, col):
                    continue
                board[row][col] = 'Q'
                backtracking(board, row + 1, n)
                board[row][col] = '.'
        backtracking(board, 0, n)
        return res

37. 解数独

题目：编写一个程序，通过填充空格来解决数独问题。
一个数独的解法需遵循如下规则： 数字 1-9 在每一行只能出现一次。 数字 1-9 在每一列只能出现一次。 数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。 空白格用 ‘.’ 表示。

题目链接：37. 解数独

class Solution:
    def solveSudoku(self, board: List[List[str]]) -> None:
        """
        Do not return anything, modify board in-place instead.
        """
        self.backtracking(board)
    
    def backtracking(self, board: List[List[str]]) -> bool:
        # 若有解，返回True；若无解，返回False
        for i in range(len(board)): # 遍历行
            for j in range(len(board[0])): # 遍历列
                # 若空格内已有数字，跳过
                if board[i][j] != '.':
                    continue
                for k in range(1, 10): # (i, j) 这个位置放k是否合适
                    if self.isValid(i, j, k, board): 
                        board[i][j] = str(k)
                        if self.backtracking(board):
                            return True
                        board[i][j] = '.' # 回溯
                return False
        return True
    
    
    def isValid(self, row: int, col: int, val: int, board: List[List[str]]) -> bool:
        # 判断同一行是否冲突
        for i in range(9):
            if board[row][i] == str(val):
                return False
        # 判断同一列是否冲突
        for j in range(9):
            if board[j][col] == str(val):
                return False

        # 判断同一九宫格是否有冲突
        start_row = (row // 3) * 3
        start_col = (col // 3) * 3
        for i in range(start_row, start_row + 3):
            for j in range(start_col, start_col + 3):
                if board[i][j] == str(val):
                    return False
        return True