代码随想录算法训练营

332.重新安排行程

题目链接：332.重新安排行程

给定一个机票的字符串二维数组 [from, to]，子数组中的两个成员分别表示飞机出发和降落的机场地点，对该行程进行重新规划排序。所有这些机票都属于一个从 JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 开始。

提示：

如果存在多种有效的行程，请你按字符自然排序返回最小的行程组合。例如，行程 ["JFK", "LGA"] 与 ["JFK", "LGB"] 相比就更小，排序更靠前
所有的机场都用三个大写字母表示（机场代码）。
假定所有机票至少存在一种合理的行程。
所有的机票必须都用一次且只能用一次。

示例 1：
输入：[["MUC", "LHR"], ["JFK", "MUC"], ["SFO", "SJC"], ["LHR", "SFO"]]
输出：["JFK", "MUC", "LHR", "SFO", "SJC"]

总体思路

直觉上来看这道题和回溯法没有什么关系，更像是图论中的深度优先搜索。

实际上确实是深搜，但这是深搜中使用了回溯的例子，在查找路径的时候，如果不回溯，怎么能查到目标路径呢。

所以我倾向于说本题应该使用回溯法，那么我也用回溯法的思路来讲解本题，其实深搜一般都使用了回溯法的思路，在图论系列中我会再详细讲解深搜。

这道题目有几个难点：

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

如何理解死循环

例如机场的循环

出发机场和到达机场也会重复的，如果在解题的过程中没有对集合元素处理好，就会死循环

记录映射关系

有多种解法，字母序靠前排在前面，让很多同学望而退步，如何该记录映射关系呢？

一个机场映射多个机场，机场之间要靠字母序排列，一个机场映射多个机场，可以使用std::unordered_map，如果让多个机场之间再有顺序的话，就是用std::map 或者std::multimap 或者 std::multiset。

如果对map 和 set 的实现机制不太了解，也不清楚为什么 map、multimap就是有序的同学，可以看这篇文章关于哈希表，你该了解这些！。

这样存放映射关系可以定义为 unordered_map<string, multiset<string>> targets 或者 unordered_map<string, map<string, int>> targets。

含义如下：

unordered_map<string, multiset> targets：unordered_map<出发机场, 到达机场的集合> targets

unordered_map<string, map<string, int>> targets：unordered_map<出发机场, map<到达机场, 航班次数>> targets

这两个结构，我选择了后者，因为如果使用unordered_map<string, multiset<string>> targets 遍历multiset的时候，不能删除元素，一旦删除元素，迭代器就失效了。

再说一下为什么一定要增删元素呢，正如开篇我给出的图中所示，出发机场和到达机场是会重复的，搜索的过程没及时删除目的机场就会死循环。

所以搜索的过程中就是要不断的删multiset里的元素，那么推荐使用unordered_map<string, map<string, int>> targets。

在遍历 unordered_map<出发机场, map<到达机场, 航班次数>> targets的过程中，可以使用"航班次数"这个字段的数字做相应的增减，来标记到达机场是否使用过了。

如果“航班次数”大于零，说明目的地还可以飞，如果“航班次数”等于零说明目的地不能飞了，而不用对集合做删除元素或者增加元素的操作。

相当于说我不删，我就做一个标记！

回溯法

本题以输入：["JFK", "KUL"], ["JFK", "NRT"], ["NRT", "JFK"]为例，抽象为树形结构如下：

回溯三部曲

确定回溯的参数

在讲解映射关系的时候，已经讲过了，使用unordered_map<string, map<string, int>> targets; 来记录航班的映射关系，定义为全局变量。

当然把参数放进函数里传进去也是可以的，我是尽量控制函数里参数的长度。

参数里还需要ticketNum，表示有多少个航班（终止条件会用上）。

// unordered_map<出发机场, map<到达机场, 航班次数>> targets

unordered_map<string, map<string, int>> targets;

bool backtracking(int ticketNum, vector<string>& result) {

注意，函数返回值使用的是bool类型因为

所以找到了这个叶子节点了直接返回.

当然本题的targets和result都需要初始化，代码如下：

for (const vector<string>& vec : tickets) {

    targets[vec[0]][vec[1]]++; // 记录映射关系

}

result.push_back("JFK"); // 起始机场

确定回溯的终止条件

拿题目中的示例为例，输入: [["MUC", "LHR"], ["JFK", "MUC"], ["SFO", "SJC"], ["LHR", "SFO"]] ，这是有4个航班，那么只要找出一种行程，行程里的机场个数是5就可以了。

所以终止条件是：我们回溯遍历的过程中，遇到的机场个数，如果达到了（航班数量+1），那么我们就找到了一个行程，把所有航班串在一起了。

代码如下：

if (result.size() == ticketNum + 1) {

    return true;

}

已经看习惯回溯法代码的同学，到叶子节点了习惯性的想要收集结果，但发现并不需要，本题的result相当于回溯算法：求组合总和！中的path，也就是本题的result就是记录路径的（就一条），在如下单层搜索的逻辑中result就添加元素了。

3. 确定回溯每一层的具体函数

在选择映射函数的时候，不能选择unordered_map<string, multiset<string>> targets，因为一旦有元素增删multiset的迭代器就会失效，当然可能有牛逼的容器删除元素迭代器不会失效，这里就不在讨论了。

可以说本题既要找到一个对数据进行排序的容器，而且还要容易增删元素，迭代器还不能失效。

所以我选择了unordered_map<string, map<string, int>> targets 来做机场之间的映射。

for (pair<const string, int>& target : targets[result[result.size() - 1]]) {

    if (target.second > 0 ) { // 记录到达机场是否飞过了

        result.push_back(target.first);

        target.second--;

        if (backtracking(ticketNum, result)) return true;

        result.pop_back();

        target.second++;

    }

}

可以看出通过unordered_map<string, map<string, int>> targets里的int字段来判断这个集合里的机场是否使用过，这样避免了直接去删元素。

完整代码

class Solution {

private:

// unordered_map<出发机场, map<到达机场, 航班次数>> targets

unordered_map<string, map<string, int>> targets;

bool backtracking(int ticketNum, vector<string>& result) {

    if (result.size() == ticketNum + 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--;

            if (backtracking(ticketNum, result)) return true;

            result.pop_back();

            target.second++;

        }

    }

    return false;

}

public:

    vector<string> findItinerary(vector<vector<string>>& tickets) {

        targets.clear();

        vector<string> result;

        for (const vector<string>& vec : tickets) {

            targets[vec[0]][vec[1]]++; // 记录映射关系

        }

        result.push_back("JFK"); // 起始机场

        backtracking(tickets.size(), result);

        return result;

    }

};

51. N皇后

题目链接：N皇后

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

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

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

示例 1：

输入：n = 4
输出：[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释：如上图所示，4 皇后问题存在两个不同的解法。

总体思路

n皇后问题是回溯算法解决的经典问题，但是用回溯解决多了组合、切割、子集、排列问题之后，遇到这种二维矩阵还会有点不知所措。

首先来看一下皇后们的约束条件：

不能同行
不能同列
不能同斜线

确定完约束条件，来看看究竟要怎么去搜索皇后们的位置，其实搜索皇后的位置，可以抽象为一棵树。

下面我用一个 3 * 3 的棋盘，将搜索过程抽象为一棵树，如图

那么我们用皇后们的约束条件，来回溯搜索这棵树，只要搜索到了树的叶子节点，说明就找到了皇后们的合理位置了。

递归三部曲

确定递归参数

依然是定义全局变量二维数组result来记录最终结果。

参数n是棋盘的大小，然后用row来记录当前遍历到棋盘的第几层了。

代码如下：

vector<vector<string>> result;

void backtracking(int n, int row, vector<string>& chessboard) {

确定递归终止条件

if (row == n) {

    result.push_back(chessboard);

    return;

}

确定每层具体函数

递归深度就是row控制棋盘的行，每一层里for循环的col控制棋盘的列，一行一列，确定了放置皇后的位置。

每次都是要从新的一行的起始位置开始搜，所以都是从0开始。

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] = '.'; // 回溯，撤销皇后

    }

}

验证棋盘是否合法

按照如下标准去重：

不能同行
不能同列
不能同斜线（45度和135度角）

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 >= 0; i--, j--) {

        if (chessboard[i][j] == 'Q') {

            return false;

        }

    }

    // 检查 135度角是否有皇后

    for(int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {

        if (chessboard[i][j] == 'Q') {

            return false;

        }

    }

    return true;

}

总体代码

class Solution {

private:

vector<vector<string>> result;

// n 为输入的棋盘大小

// row 是当前递归到棋盘的第几行了

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 >= 0; i--, j--) {

        if (chessboard[i][j] == 'Q') {

            return false;

        }

    }

    // 检查 135度角是否有皇后

    for(int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {

        if (chessboard[i][j] == 'Q') {

            return false;

        }

    }

    return true;

}

public:

    vector<vector<string>> solveNQueens(int n) {

        result.clear();

        std::vector<std::string> chessboard(n, std::string(n, '.'));

        backtracking(n, 0, chessboard);

        return result;

    }

};

37. 解数独

题目链接：解数独

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

一个数独的解法需遵循如下规则：数字 1-9 在每一行只能出现一次。数字 1-9 在每一列只能出现一次。数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。空白格用 '.' 表示。

提示：

给定的数独序列只包含数字 1-9 和字符 '.' 。
你可以假设给定的数独只有唯一解。
给定数独永远是 9x9 形式的。

总体思路

N皇后问题是因为每一行每一列只放一个皇后，只需要一层for循环遍历一行，递归来遍历列，然后一行一列确定皇后的唯一位置。

本题就不一样了，本题中棋盘的每一个位置都要放一个数字（而N皇后是一行只放一个皇后），并检查数字是否合法，解数独的树形结构要比N皇后更宽更深。

递归三部曲

确定递归参数

递归函数的返回值需要是bool类型，为什么呢？

因为解数独找到一个符合的条件（就在树的叶子节点上）立刻就返回，相当于找从根节点到叶子节点一条唯一路径，所以需要使用bool返回值。

代码如下：

bool backtracking(vector<vector<char>>& board)

确定递归终止条件

本题递归不用终止条件，解数独是要遍历整个树形结构寻找可能的叶子节点就立刻返回。

不用终止条件会不会死循环？

递归的下一层的棋盘一定比上一层的棋盘多一个数，等数填满了棋盘自然就终止（填满当然好了，说明找到结果了），所以不需要终止条件！

那么有没有永远填不满的情况呢？

这个问题我在递归单层搜索逻辑里再来讲！
确定单层搜索函数

在树形图中可以看出我们需要的是一个二维的递归（也就是两个for循环嵌套着递归）

一个for循环遍历棋盘的行，一个for循环遍历棋盘的列，一行一列确定下来之后，递归遍历这个位置放9个数字的可能性！

代码如下：（详细看注释）

bool backtracking(vector<vector<char>>& board) {

    for (int i = 0; i < board.size(); i++) {        // 遍历行

        for (int j = 0; j < board[0].size(); j++) { // 遍历列

            if (board[i][j] != '.') continue;

            for (char k = '1'; k <= '9'; k++) {     // (i, j) 这个位置放k是否合适

                if (isValid(i, j, k, board)) {

                    board[i][j] = k;                // 放置k

                    if (backtracking(board)) return true; // 如果找到合适一组立刻返回

                    board[i][j] = '.';              // 回溯，撤销k

                }

            }

            return false;                           // 9个数都试完了，都不行，那么就返回false

        }

    }

    return true; // 遍历完没有返回false，说明找到了合适棋盘位置了

}

注意这里return false的地方，这里放return false 是有讲究的。

因为如果一行一列确定下来了，这里尝试了9个数都不行，说明这个棋盘找不到解决数独问题的解！

那么会直接返回， 这也就是为什么没有终止条件也不会永远填不满棋盘而无限递归下去！

判断棋盘是否合法

判断棋盘是否合法有如下三个维度：

同行是否重复
同列是否重复
9宫格里是否重复

代码如下：

bool isValid(int row, int col, char val, vector<vector<char>>& board) {

    for (int i = 0; i < 9; i++) { // 判断行里是否重复

        if (board[row][i] == val) {

            return false;

        }

    }

    for (int j = 0; j < 9; j++) { // 判断列里是否重复

        if (board[j][col] == val) {

            return false;

        }

    }

    int startRow = (row / 3) * 3;

    int startCol = (col / 3) * 3;

    for (int i = startRow; i < startRow + 3; i++) { // 判断9方格里是否重复

        for (int j = startCol; j < startCol + 3; j++) {

            if (board[i][j] == val ) {

                return false;

            }

        }

    }

    return true;

}

总体代码

class Solution {

private:

bool backtracking(vector<vector<char>>& board) {

    for (int i = 0; i < board.size(); i++) {        // 遍历行

        for (int j = 0; j < board[0].size(); j++) { // 遍历列

            if (board[i][j] == '.') {

                for (char k = '1'; k <= '9'; k++) {     // (i, j) 这个位置放k是否合适

                    if (isValid(i, j, k, board)) {

                        board[i][j] = k;                // 放置k

                        if (backtracking(board)) return true; // 如果找到合适一组立刻返回

                        board[i][j] = '.';              // 回溯，撤销k

                    }

                }

                return false;  // 9个数都试完了，都不行，那么就返回false

            }

        }

    }

    return true; // 遍历完没有返回false，说明找到了合适棋盘位置了

}

bool isValid(int row, int col, char val, vector<vector<char>>& board) {

    for (int i = 0; i < 9; i++) { // 判断行里是否重复

        if (board[row][i] == val) {

            return false;

        }

    }

    for (int j = 0; j < 9; j++) { // 判断列里是否重复

        if (board[j][col] == val) {

            return false;

        }

    }

    int startRow = (row / 3) * 3;

    int startCol = (col / 3) * 3;

    for (int i = startRow; i < startRow + 3; i++) { // 判断9方格里是否重复

        for (int j = startCol; j < startCol + 3; j++) {

            if (board[i][j] == val ) {

                return false;

            }

        }

    }

    return true;

}

public:

    void solveSudoku(vector<vector<char>>& board) {

        backtracking(board);

    }

};

总结

回溯算法能解决如下问题：

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