回溯算法探索篇Ⅰ

ch_ah

于 2024-08-11 20:19:43 发布

阅读量261

点赞数 6

文章标签：算法数据结构 c++ 推荐算法 leetcode 深度优先

本文链接：https://blog.csdn.net/ch_ah/article/details/141110931

版权

Leetcode77——组合

题目描述：

给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。

你可以按 任何顺序 返回答案。

示例 1：

输入：n = 4, k = 2
输出：
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

算法思想：

通过递归控制循环的次数。具体请上b站大学。

算法实现：

void backtracking(vector<vector<int>> &res,vector<int> &path,int n,int k,int startindex){
        if(path.size()==k){
            res.push_back(path);
            return;
        }
        for(int i=startindex;i<=n-(k-path.size())+1;i++){
            path.push_back(i);
            backtracking(res,path,n,k,i+1);
            path.pop_back();
        }
    }
    
    vector<vector<int>> combine(int n, int k) {
        vector<vector<int>> res;
        vector<int> path;
        backtracking(res,path,n,k,1);
        return res;
    }

Leetcode216——组合总和Ⅲ

题目描述：

找出所有相加之和为 n 的 k 个数的组合，且满足下列条件：

只使用数字1到9
每个数字 最多使用一次

返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。

算法思想：

上题思想在终止条件中加入条件判断。

算法实现：

void backtracking(vector<vector<int>> &res,vector<int> &path,int k,int n,int start){
        if(accumulate(path.begin(),path.end(),0)>n) return;
        if(path.size()==k){
            if(accumulate(path.begin(),path.end(),0)==n) res.push_back(path);
            return;
        }
        for(int i=start;i<=min(9,n);i++){
            path.push_back(i);
            backtracking(res,path,k,n,i+1);
            path.pop_back();
        }
        return;
    } 
    
    vector<vector<int>> combinationSum3(int k, int n) {
        vector<vector<int>> res;
        vector<int> path;
        backtracking(res,path,k,n,1);
        return res;
    }

Leetcode17——电话号码的字母组合

题目描述：

给定一个仅包含数字 2-9 的字符串，返回所有它能表示的字母组合。答案可以按 任意顺序 返回。

给出数字到字母的映射如下（与电话按键相同）。注意 1 不对应任何字母。

示例 1：

输入：digits = "23"
输出：["ad","ae","af","bd","be","bf","cd","ce","cf"]

示例 2：

输入：digits = ""
输出：[]

算法思想：

建立好数字与字母的映射关系，再通过回溯算法解决问题。

算法实现：

//建立字母与数字的映射表
const string letter[10]={
        "",
        "",
        "abc",
        "def",
        "ghi",
        "jkl",
        "mno",
        "pqrs",
        "tuv",
        "wxyz",
    };
    void backtracking(vector<string> &res,string &path,int index,string &digits){
        if(path.size()==digits.size()){
            res.push_back(path);
            return;
        }
        int digit=digits[index]-'0';
        for(int i=0;i<letter[digit].size();i++){
            path.push_back(letter[digit][i]);
            backtracking(res,path,index+1,digits);
            path.pop_back();
        }
        return;
    }
    vector<string> letterCombinations(string digits) {
        vector<string> res;
        if(digits.size()==0) return res;
        string path;
        backtracking(res,path,0,digits);
        return res;
    }