332.重新安排行程
题目链接:332.重新安排行程
文档讲解:代码随想录
思路
关键是找到一种数据结构存储机票信息,并且能根据目的地的字母顺序进行排序。
因为要将出发地和目的地一一对应,因此可以选择使用map
结构存储机票,而一个出发地可能对应着多个目的地,所以unordered_map
的键存储出发地,值使用一个map
存储目的地,因为map
可以对键进行排序,所以可以优先选择字母顺序最小的。在map
中,键存放目的地,值使用一个int
变量存放到达该目的地的航班次数,每到达一次该目的地,次数减一,只有当次数大于0的时候才能到达此目的地,否则会造成两个地点间的循环。
unordered_map<string, map<string, int>>
代码
class Solution {
public:
vector<string> findItinerary(vector<vector<string>> &tickets) {
for (const vector<string> &vec : tickets)
targets[vec[0]][vec[1]]++; // 记录映射关系
vector<string> result;
result.push_back("JFK");
backtracking(tickets.size(), result);
return result;
}
private:
// unordered_map<出发机场, map<到达机场, 航班次数>>
unordered_map<string, map<string, int>> targets;
bool backtracking(int tickNum, vector<string> &result) {
if (result.size() == tickNum + 1)
return true;
for (pair<const string, int> &target : targets[result[result.size() - 1]]) {
if (target.second > 0) {
result.push_back(target.first);
target.second--;
bool found = backtracking(tickNum, result);
if (found)
return true;
result.pop_back();
target.second++;
}
}
return false;
}
};
51. N皇后
题目链接:51. N皇后
文档讲解:代码随想录
思路
抽象成树形结构,树的宽度(for
循环)代表遍历每列(每行),树的深度(递归)代表遍历每行(每列),如果皇后放置在此处合法,则进行递归回溯,否则跳过当次循环
判断是否合法,只需要对列和两个斜边进行检查即可。
代码
class Solution {
public:
vector<vector<string>> solveNQueens(int n) {
vector<string> chessboard(n, string(n, '.'));
backtracking(n, 0, chessboard);
return result;
}
private:
vector<vector<string>> result;
void backtracking(int n, int row, vector<string> &chessboard) {
if (row == n) {
result.push_back(chessboard);
return;
}
for (int col = 0; col < n; col++) {
if (isValid(row, col, chessboard, n)) {
chessboard[row][col] = 'Q'; // 放置皇后
backtracking(n, row + 1, chessboard);
chessboard[row][col] = '.'; // 回溯,撤销皇后
}
}
}
bool isValid(int row, int col, vector<string> chessboard, int n) {
// 检查列
for (int i = 0; i < row; i++) {
if (chessboard[i][col] == 'Q')
return false;
}
// 检查45度
for (int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
if (chessboard[i][j] == 'Q')
return false;
}
// 检查135度
for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
if (chessboard[i][j] == 'Q')
return false;
}
return true;
}
};
37. 解数独
题目链接:37. 解数独
文档讲解:代码随想录
思路
因为要对每个空格子进行1-9的判断,所以要使用两层for
循环遍历行和列,递归遍历1-9。
函数需要带有返回值,当找到一种答案时立刻返回。
注意return false
的位置,1-9都判断完后没有符合条件的,则返回false
并进行回溯并修改之前确定的值。
代码
class Solution {
public:
void solveSudoku(vector<vector<char>> &board) {
backtracking(board);
}
private:
bool backtracking(vector<vector<char>> &board) {
for (int i = 0; i < board.size(); i++) { // 遍历行
for (int j = 0; j < board.size(); j++) { // 遍历列
if (board[i][j] != '.')
continue;
for (char k = '1'; k <= '9'; k++) {
if (isValid(board, i, j, k)) {
board[i][j] = k;
bool found = backtracking(board);
if (found)
return true;
board[i][j] = '.';
}
}
return false;
}
}
return true;
}
bool isValid(vector<vector<char>> board, int row, int col, char num) {
// 检查行
for (int i = 0; i < board.size(); i++)
if (board[row][i] == num)
return false;
// 检查列
for (int i = 0; i < board.size(); i++)
if (board[i][col] == num)
return false;
// 检查九宫格
int row_grid = row / 3;
int col_grid = col / 3;
for (int i = 3 * row_grid; i < 3 * row_grid + 3; i++) {
for (int j = 3 * col_grid; j < 3 * col_grid + 3; j++) {
if (board[i][j] == num)
return false;
}
}
return true;
}
};