代码随想录算法训练营|day32
332.重新安排行程
题目描述:给你一份航线列表 tickets ,其中 tickets[i] = [fromi, toi] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。
所有这些机票都属于一个从 JFK(肯尼迪国际机场)出发的先生,所以该行程必须从 JFK 开始。如果存在多种有效的行程,请你按字典排序返回最小的行程组合。
例如,行程 [“JFK”, “LGA”] 与 [“JFK”, “LGB”] 相比就更小,排序更靠前。
假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。
题目难度:困难
时间复杂度:O(n*2n);空间复杂度:O(n)
下面展示 代码
:
class Solution:
def backtracking(self, tickets, used, cur, path):
# 注:此处totalnum为所有航班的经过的机场个数,因为所有航班都必须经过且只经过一次
if len(path) == len(tickets) + 1:
return True
for i, dest in enumerate(tickets):
if used[i] == False and dest[0] == cur:
path.append(dest[1]) # 已去过的机场
used[i] = True # 该路线已使用
if self.backtracking(tickets, used, dest[1], path):
return True
path.pop()
used[i] = False
return False # 必须加上,否则超时
def findItinerary(self, tickets: List[List[str]]) -> List[str]:
tickets.sort()
used = [False] * len(tickets)
path = ['JFK']
self.backtracking(tickets, used, 'JFK', path)
return path[:]
51. N皇后
题目描述:按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。
n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n ,返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。
题目难度:困难
时间复杂度:O(n!);空间复杂度:O(n)
代码
如下
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: int, row: int, chessboard: List[str], result: List[List[str]]) -> None:
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: int, col: int, chessboard: List[str]) -> bool:
# 检查列
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. 解数独
题目描述:编写一个程序,通过填充空格来解决数独问题。
数独的解法需 遵循如下规则:
数字 1-9 在每一行只能出现一次。
数字 1-9 在每一列只能出现一次。
数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。(请参考示例图)
数独部分空格内已填入了数字,空白格用 ‘.’ 表示
题目难度:困难
时间复杂度:O(9*n!);空间复杂度:O(n)
代码
如下
class Solution:
# 思路:一个for循环遍历棋盘的行,一个for循环遍历棋盘的列,一行一列确定下来之后,递归遍历这个位置放9个数字的可能性!
def solveSudoku(self, board: List[List[str]]) -> None:
self.backtracking(board) # 回溯求解
def backtracking(self, board):
# 行
for row in range(len(board)):
# 列
for col in range(len(board[0])):
if board[row][col] != '.':
continue
for k in range(1,10):
if self.isValid(row, col, k, board):
board[row][col] = str(k)
if self.backtracking(board):
return True
board[row][col] = '.'
return False
return True
def isValid(self, row: int, col: int, val: int, board: List[str]) -> bool:
# 检查行
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 # 当前位置合法