代码随想录训练营Day 26|Python|Leetcode|● 332.重新安排行程 ● 51. N皇后 ● 37. 解数独

332.重新安排行程

给你一份航线列表 tickets ，其中 tickets[i] = [fromi, toi] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。

所有这些机票都属于一个从 JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 开始。如果存在多种有效的行程，请你按字典排序返回最小的行程组合。

• 例如，行程 ["JFK", "LGA"] 与 ["JFK", "LGB"] 相比就更小，排序更靠前。

假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。

解题思路：

难点：来源代码随想录 (programmercarl.com)

1. 一个行程中，如果航班处理不好容易变成一个圈，成为死循环
2. 有多种解法，字母序靠前排在前面，让很多同学望而退步，如何该记录映射关系呢 ？
3. 使用回溯法（也可以说深搜） 的话，那么终止条件是什么呢？
4. 搜索的过程中，如何遍历一个机场所对应的所有机场。

首先以目的地为排序标准进行字母排序，设置dictionary存储出发地和目的地，出发地作为key，目的地作为value，进行dfs遍历key的每个value，while loop，将value作为下一个参数输入递归。result倒叙记录每一个value。

class Solution:
    def findItinerary(self, tickets: List[List[str]]) -> List[str]:
        adj = {}
        #sort the tickets
        tickets.sort(key=lambda x:x[1])
        for u,v in tickets:
            if u in adj:
                adj[u].append(v)
            else:adj[u] = [v]
        result = []
        def dfs(s):
            while s in adj and len(adj[s]) > 0:
                v = adj[s].pop(0)
                dfs(v)
            result.append(s)
        dfs('JFK')
        return result[::-1]

51. N皇后

按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 'Q' 和 '.' 分别代表了皇后和空位。

解题思路：

本题难点在于理解非一维数组转换到高一维数组，但大体思路与其他回溯题目一致。

输入参数：def backtracking(chessboard, n, row) chessboard记录当前路径，和path功能相同；n表示chessboard长度，为了方便处理边界；row为行数，与start_index相同

停止条件：if row == n: result.append(chessboard)

递归逻辑：for i in range(n): if is_valid(chessboard[row][i]): chessboard[row][i] = 'Q', 递归, chessboard[row][i] = '.' 

class Solution:
    def solveNQueens(self, n: int) -> List[List[str]]:
        result = []
        chessboard = ['.'*n for _ in range(n)]
        def backtracking(chessboard, n, row):
            if row == n:
                print(chessboard[:])
                result.append(chessboard[:])
                return 
            for i in range(n):
                # 检查每一列
                if is_valid(row, i, chessboard):
                    # 每一行添加Q
                    chessboard[row] = chessboard[row][:i] + 'Q' + chessboard[row][i+1:] 
                    backtracking(chessboard,n, row+1)
                    chessboard[row] = chessboard[row][:i] + '.' + chessboard[row][i+1:]
                    
        def is_valid(row, col, chessboard):
            #check vertically
            for i in range(row):
                if chessboard[i][col] == 'Q':
                    return False
            #check 45 degree
            i, j = row-1, col-1#check the previous rows and cols
            while i>=0 and j >= 0:
                if chessboard[i][j] == 'Q':
                    return False
                i -= 1
                j -= 1
            #check 135 degree
            m, n = row-1, col+1
            while m>= 0 and n < len(chessboard):
                if chessboard[m][n] == 'Q':
                    return False
                m-=1
                n+=1
            return True
        backtracking(chessboard, n, 0)
        return result

37. 解数独 

编写一个程序，通过填充空格来解决数独问题。

数独的解法需 遵循如下规则：

1. 数字 1-9 在每一行只能出现一次。
2. 数字 1-9 在每一列只能出现一次。
3. 数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。（请参考示例图）

数独部分空格内已填入了数字，空白格用 '.' 表示。

解题思路：

本题使用三个for loop，分别用来记录横向搜索，纵向搜索，以及1-9数字选择。如果所在值已经有数字填充，则跳过该位置。对于可填充的位置，如果is_valid为True，则直接填入当前数字，进行下一层递归，如果这些数字都可以完美填入，说明目前所i安茹的所有数字是满足要求的，不用再检查其他数字，直接return True即可。

class Solution:
    def solveSudoku(self, board: List[List[str]]) -> None:
        """
        Do not return anything, modify board in-place instead.
        """
        def backtracking(board):
            for i in range(len(board)):#row
                for j in range(len(board[0])):
                    if board[i][j] != '.':
                        continue
                    for k in range(1, 10):
                        if is_valid(i, j, k, board):
                            board[i][j] = str(k)
                            if backtracking(board):
                                return True #no need to check the other number
                            board[i][j] = '.'
                    #若1-9都不满足条件，返回False
                    return False
            return True
        
        def is_valid(row, col, val, board):
            for i in range(9):
                if board[row][i] == str(val):
                    return False
            for i in range(9):
                if board[i][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

        backtracking(board)