代码随想录算法训练营第三十天| 332.重新安排行程、51.N皇后 、37.解数独
代码随想录算法训练营第三十天| 332.重新安排行程、51.N皇后 、37.解数独
题目
332.重新安排行程
给你一份航线列表 tickets ，其中 tickets[i] = [fromi, toi] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。
所有这些机票都属于一个从 JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 开始。如果存在多种有效的行程，请你按字典排序返回最小的行程组合。
- 例如，行程
["JFK", "LGA"]与["JFK", "LGB"]相比就更小，排序更靠前。
假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。
from collections import defaultdict
class Solution:
def findItinerary(self, tickets: List[List[str]]) -> List[str]:
targets = defaultdict(list)
for ticket in tickets:
targets[ticket[0]].append(ticket[1])
for cur in targets:
targets[cur].sort()
curPath = ['JFK']
self.backstracking(targets, len(tickets), curPath)
return curPath
def backstracking(self, targets, ticketNum, curPath):
if ticketNum+1 == len(curPath):return True
cur = curPath[-1]
destinations = targets[cur]
for i, destination in enumerate(destinations):
targets[cur].pop(i)
curPath.append(destination)
if self.backstracking(targets, ticketNum, curPath):return True
targets[cur].insert(i, destination)
curPath.pop()
return False
题目
51.N皇后
按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。
n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。
给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 'Q' 和 '.' 分别代表了皇后和空位。
class Solution:
def solveNQueens(self, n: int) -> List[List[str]]:
res = []
chessboard = ["." * n for _ in range(n)]
self.backtracking(n, 0, chessboard, res)
return [[ ''.join(row) for row in solution] for solution in res]
def backtracking(self, n, row, chessboard, res):
if row == n:
res.append(chessboard[:])
return
for col in range(n):
if self.isValid(row, col, n, chessboard):
chessboard[row] = chessboard[row][:col] + 'Q' + chessboard[row][col+1:]
self.backtracking(n, row+1, chessboard, res)
chessboard[row] = chessboard[row][:col] + '.' + chessboard[row][col+1:]
def isValid(self, row, col, n, chessboard):
for i in range(row):
if chessboard[i][col] == 'Q':return False
i, j = row-1, col-1
while i >= 0 and j >= 0:
if chessboard[i][j] == 'Q':return False
i -= 1
j -= 1
i, j = row-1, col+1
while i >= 0 and j < n:
if chessboard[i][j] == 'Q':return False
i -= 1
j += 1
return True
题目
37.解数独
编写一个程序，通过填充空格来解决数独问题。
数独的解法需 遵循如下规则：
- 数字
1-9在每一行只能出现一次。 - 数字
1-9在每一列只能出现一次。 - 数字
1-9在每一个以粗实线分隔的3x3宫内只能出现一次。（请参考示例图）
数独部分空格内已填入了数字，空白格用 '.' 表示。
class Solution:
def solveSudoku(self, board: List[List[str]]) -> None:
self.backstracking(board)
def backstracking(self, board):
for row in range(len(board)):
for col in range(len(board[row])):
if board[row][col] != '.':continue
for k in range(1, 10):
if self.isValid(row, col, str(k), board):
board[row][col] = str(k)
if self.backstracking(board):return True
board[row][col] = '.'
return False
return True
def isValid(self, row, col, k, board):
# 检查行
for i in range(9):
if board[i][col] == k:return False
# 检查列
for j in range(9):
if board[row][j] == k:return False
newRow, newCol = (row//3)*3, (col//3)*3
for i in range(newRow, newRow+3):
for j in range(newCol, newCol+3):
if board[i][j] == k:return False
return True