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

最新推荐文章于 2024-07-10 16:47:58 发布

丁希希哇

最新推荐文章于 2024-07-10 16:47:58 发布

阅读量919

点赞数 22

分类专栏：力扣算法刷题文章标签：算法面试 python 力扣

本文链接：https://blog.csdn.net/weixin_47748259/article/details/136119069

版权

力扣算法刷题专栏收录该内容

69 篇文章 4 订阅

订阅专栏

系列文章目录

文章目录

系列文章目录
332.重新安排行程
51. N皇后
37. 解数独
总结

332.重新安排行程

题目链接： 332.重新安排行程
题目内容： 给你一份航线列表 tickets ，其中 tickets[i] = [fromi, toi] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。所有这些机票都属于一个从 JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 开始。如果存在多种有效的行程，请你按字典排序返回最小的行程组合。例如，行程 [“JFK”, “LGA”] 与 [“JFK”, “LGB”] 相比就更小，排序更靠前。假定所有机票至少存在一种合理的行程。且所有的机票必须都用一次且只能用一次。

思路：

一个行程中，如果航班处理不好容易变成一个圈，成为死循环
有多种解法，字母序靠前排在前面，需要正确记录映射关系
回溯法的树形结构绘制
- 递归函数参数：result数组，path数组，targetnum：（表示有多少个航班，用于终止条件）
- 递归终止条件：遇到的机场个数达到了航班数量+1，就说明找到了一个把所有航班串在一起的行程
- 单层搜索的逻辑：既要找到一个对数据进行排序的容器，而且还要容易增删元素，迭代器还不能失效。

from collections import defaultdict

class Solution:
    def findItinerary(self, tickets):
        targets = defaultdict(list)  # 创建默认字典，用于存储机场映射关系
        for ticket in tickets:
            targets[ticket[0]].append(ticket[1])  # 将机票输入到字典中
        
        for key in targets:
            targets[key].sort(reverse=True)  # 对到达机场列表进行字母逆序排序
        
        result = []
        self.backtracking("JFK", targets, result)  # 调用回溯函数开始搜索路径
        return result[::-1]  # 返回逆序的行程路径
    
    def backtracking(self, airport, targets, result):
        while targets[airport]:  # 当机场还有可到达的机场时
            next_airport = targets[airport].pop()  # 弹出下一个机场
            self.backtracking(next_airport, targets, result)  # 递归调用回溯函数进行深度优先搜索
        result.append(airport)  # 将当前机场添加到行程路径中

51. N皇后

题目链接： 51.N皇后
题目内容： 按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。给你一个整数 n ，返回所有不同的 n 皇后问题的解决方案。每一种解法包含一个不同的 n 皇后问题的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。
视频讲解： 这就是传说中的N皇后？回溯算法安排！| LeetCode：51.N皇后

result是三维数组

class Solution:
    def solveNQueens(self, n: int) -> List[List[str]]:
        result=[]
        chessboard=['.' * n for _ in range(n)]
        self.backtracking(n,0,chessboard,result)
        return [[''.join(row) for row in solution] for solution in result]

    def backtracking(self,n,row,chessboard,result):
        if row == n:
            result.append(chessboard[:])
            return
        for col in range(n):
            if self.isValid(row,col,chessboard):
                chessboard[row]=chessboard[row][:col]+'Q'+chessboard[row][col+1:]
                self.backtracking(n,row+1,chessboard,result)
                chessboard[row]=chessboard[row][:col]+'.'+chessboard[row][col+1:]

    def isValid(self,row,col,chessboard):
        #检查列
        for i in range(row):
            if chessboard[i][col] == 'Q':
                return False #该列已经存在皇后，不合法
        #检查45度角
        i,j=row-1,col-1
        while i>=0 and j>=0:
            if chessboard[i][j] == 'Q':
                return False #左上方向已经存在皇后，不合法
            i-=1
            j-=1
        #检查135度角
        i,j=row-1,col+1
        while i>=0 and j<len(chessboard):
            if chessboard[i][j] == 'Q':
                return False #左上方向已经存在皇后，不合法
            i-=1
            j+=1        
        return True

37. 解数独

题目链接： 37.解数独
题目内容： 编写一个程序，通过填充空格来解决数独问题。数独的解法需遵循如下规则：数字 1-9 在每一行只能出现一次。数字 1-9 在每一列只能出现一次。数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。数独部分空格内已填入了数字，空白格用 ‘.’ 表示。
视频讲解： 回溯算法二维递归？解数独不过如此！| LeetCode：37. 解数独

核心思路：二维递归+回溯
棋盘的每一个位置都要放一个数字（而N皇后是一行只放一个皇后），并检查数字是否合法，解数独的树形结构要比N皇后更宽更深。

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):
        for i in range(len(board)):
            for j in range(len(board[0])):
                if board[i][j] != '.':
                    continue
                for k in range(1,10):
                    if self.is_valid(i,j,k,board):
                        board[i][j] = str(k)
                        if self.backtracking(board):
                            return True
                        board[i][j] = '.'
                return False
        return True

    def is_valid(self,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

总结

组合问题：N个数里面按一定规则找出k个数的集合
排列问题：N个数按一定规则全排列，有几种排列方式
切割问题：一个字符串按一定规则有几种切割方式
子集问题：一个N个数的集合里有多少符合条件的子集
棋盘问题：N皇后，解数独等等

在这里插入图片描述

丁希希哇

关注

22
点赞
踩
17

收藏

觉得还不错? 一键收藏
0
评论
代码随想录算法训练营第二十六天|332.重新安排行程，51. N皇后，37. 解数独，总结

组合问题：N个数里面按一定规则找出k个数的集合排列问题：N个数按一定规则全排列，有几种排列方式切割问题：一个字符串按一定规则有几种切割方式子集问题：一个N个数的集合里有多少符合条件的子集棋盘问题：N皇后，解数独等等。
复制链接

扫一扫