Day24：代码随想录算法训练营第二十四天|回溯算法基础 1.力扣332 重新安排行程 2.力扣51 N皇后 3.力扣37 解数独

最新推荐文章于 2024-07-23 17:36:37 发布

当代大学生高级码农

最新推荐文章于 2024-07-23 17:36:37 发布

阅读量74

点赞数

文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/love_computer123/article/details/129908546

版权

力扣332 重新安排行程

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

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

示例 1：

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

示例 2：

输入：[["JFK","SFO"],["JFK","ATL"],["SFO","ATL"],["ATL","JFK"],["ATL","SFO"]]
输出：["JFK","ATL","JFK","SFO","ATL","SFO"]
解释：另一种有效的行程是 ["JFK","SFO","ATL","JFK","ATL","SFO"]。但是它自然排序更大更靠后。

思路：
这道题主要是不知道如何选取容器来表示每一个结点的形式，当如果是组合问题时，题目给一个数组，由i就可以控制深度和广度的遍历。
但是这道题需要先对这些机票组做一个映射，意思就是把他存入到一个“类似数组的地方”，方便我们进行深度和广度的搜索。

这里主选map，利用一个unordered_map<string, map<string, int>>,前一个string存的是出发机场，后一个string存的是到达机场，int表示航班次数，如果“航班次数”大于零，说明目的地还可以飞，如果“航班次数”等于零说明目的地不能飞了，而不用对集合做删除元素或者增加元素的操作。

对于宽度的遍历采用可以下这个for循环，因为targets里面的第一个元素是出发机场，我们每次要找的都是到达机场，因此这里遍历的是以result最后一个机场作为出发机场对应的到达机场，由于map本身是有序的，无形中使出发机场对应的多个到达机场都有了顺序，每次压入的是这多个到达机场的第一个，也就是字符最小的那个到达机场排前面。

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

在这里插入图片描述
代码如下：

class Solution {
private:
// unordered_map<出发机场, map<到达机场, 航班次数>> targets
unordered_map<string, map<string, int>> targets;
bool backtracking(int ticketNum, vector<string>& result) {
//当result里的机场数等于了票数加1，那就是说明所有机场都飞过了
    if (result.size() == ticketNum + 1) {
        return true;
    }
    //从出发机场找对应储存在map里的到达机场
    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;
            //如果找到了逐级向上返回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 皇后问题的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

示例 1：
在这里插入图片描述输入：n = 4
输出：[[“.Q…”,“…Q”,“Q…”,“…Q.”],[“…Q.”,“Q…”,“…Q”,“.Q…”]]
解释：如上图所示，4 皇后问题存在两个不同的解法。

思路：
一个是判断可以为皇后位，一个是二维的填充问题。
抽象树的宽度是当前行的4个位置，深度是棋盘的长度。
那就是每一行找一个，然后递归到下一行找，找不到或是找到了回溯上一行找下一个位置。

代码如下：

class Solution {
public:
vector<vector<string>> result;
    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;
            }
        }
        for(int i = row -1, j = col - 1; i >= 0 && j >= 0  ; i--, j--){//检查45度是否有Q
                if(chessboard[i][j] == 'Q'){
                    return false;
                }
        }
       for(int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
        if (chessboard[i][j] == 'Q') {
            return false;
        }
    }
    return true;
}
    void backtracking(int n, int row, vector<string>& chessboard){
        if(row == n){
            result.push_back(chessboard);
            return;
        }
        for(int i = 0; i < n; i++){
            if(isValid(row, i, chessboard, n)){
                chessboard[row][i] = 'Q';
                backtracking(n, row + 1, chessboard);
                //下一层如果四个位置均不行或是找到了结果，那就把皇后位重新置位"."
                chessboard[row][i] = '.';
            }else continue;
    }
}
    vector<vector<string>> solveNQueens(int n) {
        vector<string> chessboard(n,string(n, '.'));
        backtracking(n, 0, chessboard);
        return result;
    }
};

力扣37 解数独

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

一个数独的解法需遵循如下规则：数字 1-9 在每一行只能出现一次。数字 1-9 在每一列只能出现一次。数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。空白格用 ‘.’ 表示。
在这里插入图片描述输入：board = [[“5”,“3”,“.”,“.”,“7”,“.”,“.”,“.”,“.”],[“6”,“.”,“.”,“1”,“9”,“5”,“.”,“.”,“.”],[“.”,“9”,“8”,“.”,“.”,“.”,“.”,“6”,“.”],[“8”,“.”,“.”,“.”,“6”,“.”,“.”,“.”,“3”],[“4”,“.”,“.”,“8”,“.”,“3”,“.”,“.”,“1”],[“7”,“.”,“.”,“.”,“2”,“.”,“.”,“.”,“6”],[“.”,“6”,“.”,“.”,“.”,“.”,“2”,“8”,“.”],[“.”,“.”,“.”,“4”,“1”,“9”,“.”,“.”,“5”],[“.”,“.”,“.”,“.”,“8”,“.”,“.”,“7”,“9”]]
输出：[[“5”,“3”,“4”,“6”,“7”,“8”,“9”,“1”,“2”],[“6”,“7”,“2”,“1”,“9”,“5”,“3”,“4”,“8”],[“1”,“9”,“8”,“3”,“4”,“2”,“5”,“6”,“7”],[“8”,“5”,“9”,“7”,“6”,“1”,“4”,“2”,“3”],[“4”,“2”,“6”,“8”,“5”,“3”,“7”,“9”,“1”],[“7”,“1”,“3”,“9”,“2”,“4”,“8”,“5”,“6”],[“9”,“6”,“1”,“5”,“3”,“7”,“2”,“8”,“4”],[“2”,“8”,“7”,“4”,“1”,“9”,“6”,“3”,“5”],[“3”,“4”,“5”,“2”,“8”,“6”,“1”,“7”,“9”]]
解释：输入的数独如上图所示，唯一有效的解决方案如下所示：

在这里插入图片描述答案被标成红色。

提示：

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

思路：
这道题很难，建议看代码理解

代码如下:

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);
    }
};

当代大学生高级码农

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
Day24：代码随想录算法训练营第二十四天|回溯算法基础 1.力扣332 重新安排行程 2.力扣51 N皇后 3.力扣37 解数独

Day24：代码随想录算法训练营第二十四天|回溯算法基础 1.力扣332 重新安排行程 2.力扣51 N皇后 3.力扣37 解数独
复制链接

扫一扫