Leetcode算法训练日记 | day30

专题九  贪心算法

一、重新安排行程

1.题目

Leetcode:第 332 题

给你一份航线列表 tickets ,其中 tickets[i] = [fromi, toi] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。

所有这些机票都属于一个从 JFK(肯尼迪国际机场)出发的先生,所以该行程必须从 JFK 开始。如果存在多种有效的行程,请你按字典排序返回最小的行程组合。

  • 例如,行程 ["JFK", "LGA"] 与 ["JFK", "LGB"] 相比就更小,排序更靠前。

假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。

示例 1:

输入:tickets = [["MUC","LHR"],["JFK","MUC"],["SFO","SJC"],["LHR","SFO"]]
输出:["JFK","MUC","LHR","SFO","SJC"]

示例 2:

输入:tickets = [["JFK","SFO"],["JFK","ATL"],["SFO","ATL"],["ATL","JFK"],["ATL","SFO"]]
输出:["JFK","ATL","JFK","SFO","ATL","SFO"]
解释:另一种有效的行程是 ["JFK","SFO","ATL","JFK","ATL","SFO"] ,但是它字典排序更大更靠后。

2.解题思路

使用回溯算法来解决路径问题。findItinerary 函数负责初始化并开始回溯过程。backtracking 函数是回溯算法的核心,它尝试在每个位置添加一个城市,并递归地继续处理后续的城市。通过这种方式,backtracking 函数能够找到一条有效的旅行路径。targets 映射用于存储每个出发城市到到达城市的机票数量,这个映射在 findItinerary 函数中被构建,并在 backtracking 函数中被用来检查是否存在可用的机票。通过递归和回溯来搜索所有可能的解决方案。当找到一个满足条件的解时,递归返回 true,否则尝试其他可能的路径。

3.实现代码

#include <iostream>
#include <vector>
#include <map>
#include <unordered_map>
using namespace std;

class Solution {
public:
    // 使用 unordered_map 存储目标城市,其中 key 为出发城市,value 为一个 map,其 key 为到达城市,value 为机票数量
    unordered_map<string, map<string, int>> targets;

    // backtracking 函数实现回溯算法,尝试找出一条有效的旅行路径
    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]]) {
            // 如果从当前城市到 target.first 的机票数量大于 0
            if (target.second > 0) { 
                result.push_back(target.first);// 将 target.first 添加到结果路径中
                target.second--;// 减少 target.first 的机票数量
                if (backtracking(ticketNum, result)) return true; // 递归调用 backtracking 函数,尝试继续添加路径  
                result.pop_back();// 如果当前路径不可行,回溯:移除最后一个添加的城市
                target.second++;// 恢复机票数量
            }
        }
        return false;// 如果所有可能的路径都已经尝试过,但没有找到可行的路径,则返回 false
    }

    // findItinerary 函数用于找出一条最短的旅行路径
    vector<string> findItinerary(vector<vector<string>>& tickets) {
        targets.clear(); // 清空 targets,为新的机票数据做准备
        vector<string> result;
        // 统计每个出发城市到到达城市的机票数量
        for (const vector<string>& vec : tickets) {
            targets[vec[0]][vec[1]]++;
        }
        result.push_back("JFK");// 将起点城市 "JFK" 添加到结果路径中 
        backtracking(tickets.size(), result);// 从起点城市开始,使用 backtracking 函数尝试找出一条旅行路径
        return result; // 返回找到的旅行路径
    }
};

//测试
int main()
{
    Solution p;
    vector<vector<string>>tickets = { {"MUC", "LHR"} ,{"JFK", "MUC"},{"SFO", "SJC"},{"LHR", "SFO"} };
   vector<string> result;
    vector<int>nums = { 1,2,2 };
    result = p.findItinerary(tickets);
    cout << "行程:" << endl;
    for (auto& ans : result) {
        cout << "[";
        for (auto& i : ans) {
            cout << i << " ";
        }
        cout << "] ," ;
    }
    printf("\b \n");
    return 0;
}

二、N皇后

1.题目

Leetcode:第 51 题

按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

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

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

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

示例 1:

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

示例 2:

输入:n = 1
输出:[["Q"]]
2.解题思路

使用回溯算法来解决棋盘问题。solveNQueens 函数负责初始化棋盘并开始回溯过程。backtracking 函数是回溯算法的核心,它尝试在每个位置放置皇后,并递归地继续处理后续的行。isValid 函数用于检查当前位置是否安全,即没有受到其他皇后的攻击。backtracking 函数通过递归尝试每一行的每一个位置来放置皇后。如果找到一个安全的位置,它将继续递归到下一行。如果这一行的所有位置都尝试过了,它将回溯到上一行,改变上一行皇后的位置。isValid 函数通过检查当前行的列、左上到右下的对角线、右上到左下的对角线是否有皇后来确定当前位置是否安全。这种方法可以找到所有可能的解决方案,即所有安全的皇后放置方式。

3.实现代码
#include <iostream>
#include <vector>
using namespace std;

class Solution {
public:
    vector<vector<string>> result;// 定义一个二维字符串数组用于存储所有可能的棋盘布局结果

    // backtracking 函数实现回溯算法,尝试放置每一行的皇后
    void backtracking(int n, int row, vector<string>& chessboard) {
        // 如果当前行数等于 n,说明已经放置完所有行的皇后,找到了一个解决方案
        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);// 递归调用 backtracking 函数,尝试放置下一行的皇后
                chessboard[row][col] = '.';// 移除当前位置的皇后,回溯以便尝试其他位置
            }
        }
    }

    // isValid 函数用于检查当前行和列是否是安全的,即该位置没有受到其他皇后的攻击
    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; // 如果同一列有皇后,则返回 false
            }
        }
        // 检查左上到右下的对角线是否有其他皇后
        for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
            if (chessboard[i][j] == 'Q') {
                return false; // 如果对角线有皇后,则返回 false
            }
        }
        // 检查右上到左下的对角线是否有其他皇后
        for (int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
            if (chessboard[i][j] == 'Q') {
                return false; // 如果对角线有皇后,则返回 false
            }
        }
        return true;// 如果当前位置安全,则返回 true
    }

    // solveNQueens 函数用于开始解决 N 皇后问题
    vector<vector<string>> solveNQueens(int n) {      
        result.clear();// 清空结果集,为新的棋盘布局做准备
        vector<string> chessboard(n, string(n, '.')); // 初始化棋盘,所有位置先设为 '.'
        backtracking(n, 0, chessboard);// 从第一行开始放置皇后
        return result;// 返回所有可能的棋盘布局结果
    }
};

//测试
int main()
{
    Solution s;
    int n;
    vector<vector<string>>result;
    cout << "请输入 n : ";
    cin >> n;
    result = s.solveNQueens(n);
    cout << "所有的解法有:" << endl;
    for (int i = 0; i < result.size(); i++) {
        cout << "第"<<i+1<<"种解法" << endl;
        for (int j = 0; j < n; j++) {
            cout << result[i][j] << "  " << endl;
        }
        cout << endl;
    }
    cout <<endl;
    return 0;
}

三、解数独

1.题目

Leetcode:第 37 题

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

数独的解法需 遵循如下规则

  1. 数字 1-9 在每一行只能出现一次。
  2. 数字 1-9 在每一列只能出现一次。
  3. 数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。(请参考示例图)

数独部分空格内已填入了数字,空白格用 '.' 表示。

示例 1:

输入: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"]]
解释:输入的数独如上图所示,唯一有效的解决方案如下所示:

2.解题思路

使用回溯算法来解决数独问题。solveSudoku 函数负责开始回溯过程。backtracking 函数是回溯算法的核心,它尝试在每个空格填入数字,并递归地继续处理后续的空格。isValid 函数用于检查当前位置填入某个数字后数独板是否仍然有效。backtracking 函数通过递归尝试每个空格的每个可能的数字来解决数独问题。如果找到一个数字使得数独板有效,它将继续递归到下一个空格。如果当前空格的所有尝试都失败了,它将回溯到上一个空格,改变上一个空格的数字。isValid 函数通过检查当前行、当前列以及当前单元格所在的 3x3 子网格是否有重复的数字来确定当前位置是否有效。这种方法可以找到数独问题的解决方案,即使在复杂的数独谜题中也能有效地找到答案。

3.实现代码
#include <iostream>
#include <vector>
using namespace std;

class Solution {
public:
    // backtracking 函数实现回溯算法,尝试填入数独板的每个空格
    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] == '.') {
                    // 尝试填入数字 1 到 9
                    for (char k = '1'; k <= '9'; k++) {
                        // 如果填入当前数字 k 后数独板仍然有效
                        if (isValid(i, j, k, board)) {
                            // 在当前单元格填入数字 k
                            board[i][j] = k;
                            // 递归调用 backtracking 函数,尝试填入下一个空格
                            if (backtracking(board)) {
                                return true; // 如果找到一个解决方案,则返回 true
                            }
                            // 如果当前数字 k 不适用,回溯:将当前单元格重置为空
                            board[i][j] = '.';
                        }
                    }
                    return false;// 如果没有数字可以填入当前单元格,则返回 false
                }
            }
        }
        return true;// 如果所有单元格都已经填入数字并且数独板有效,则返回 true
    }

    // isValid 函数检查在给定的行和列填入某个值后数独板是否仍然有效
    bool isValid(int row, int col, char val, vector<vector<char>>& board) {
        // 检查当前行是否已包含要填入的值 val
        for (int i = 0; i < 9; i++) {
            if (board[row][i] == val) {
                return false; // 如果已包含,则返回 false
            }
        }
        // 检查当前列是否已包含要填入的值 val
        for (int j = 0; j < 9; j++) {
            // 数独规则中,即使单元格为空('.'),列中也不能有重复的数字
            if (board[j][col] == val) {
                return false; // 如果已包含,则返回 false
            }
        }

        // 计算当前单元格所在的 3x3 子网格的起始行和列
        int startRow = (row / 3) * 3;
        int startCol = (col / 3) * 3;
        // 检查当前 3x3 子网格是否已包含要填入的值 val
        for (int i = startRow; i < startRow + 3; i++) {
            for (int j = startCol; j < startCol + 3; j++) {
                if (board[i][j] == val) {
                    return false; // 如果已包含,则返回 false
                }
            }
        }
        // 如果当前单元格填入 val 后不违反数独规则,则返回 true
        return true;
    }

    // solveSudoku 函数用于解决数独问题
    void solveSudoku(vector<vector<char>>& board) {
        backtracking(board);// 调用 backtracking 函数尝试填入数独板
    }
};

//测试
int main()
{
    Solution s;
    vector<vector<char>> 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'} };
    s.solveSudoku(board);
    cout << "数独的其中一种解法:" << endl;
    for (int i = 0; i < board.size(); i++) {
        for (int j = 0; j < board[0].size(); j++) {
            cout << board[i][j] << "  ";
        }
        cout << endl;
    }
    cout << endl;
    cout << endl;
    return 0;
}

 ps:以上皆是本人在探索算法旅途中的浅薄见解,诚挚地希望得到各位的宝贵意见与悉心指导,若有不足或谬误之处,还请多多指教。

  • 28
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值