代码随想录算法训练营第25天 | 回溯算法2

原创已于 2023-02-19 22:53:14 修改 · 321 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法 #leetcode #数据结构

于 2023-02-09 22:31:39 首次发布

代码随想录专栏收录该内容

11 篇文章

订阅专栏

文章介绍了如何使用回溯算法解决LeetCode中的两道题目：寻找相加等于目标值的数字组合（限制数字使用次数）和电话号码的字母组合问题。通过示例代码展示了回溯法在解决这类问题中的应用。

216.组合总合III

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

只使用数字1到9每个数字
最多使用一次返回所有可能的有效组合的列表。该列表不能包含相同的组合两次，组合可以以任何顺序返回。
链接： https://leetcode.cn/problems/combination-sum-iii
思路：
vector<int> path 存放满足条件的结果
vector<vector<int>> res 存放满足条件的结果的集合
回溯算法三部曲
①传入参数：int k, int n, int startIndex, int sum
②终止条件：path.size() == k 而且sum == n
③遍历过程：for从startIndex开始遍历到9 ，如果和为n，就把path存入res，然后回溯；不为n直接回溯。回溯：sum -= i, path.pop_back();

class Solution {
private:
    vector<int> path;
    vector<vector<int>> res;
    void backtracking(int k, int n, int startIndex, int sum){
        if(path.size() == k){
            if(sum == n){
                res.push_back(path);
                return;
            }
        }
        for(int i = startIndex; i <= 9; ++i){
            sum += i;
            path.push_back(i); 
            backtracking(k, n, i + 1, sum);
            sum -= i;
            path.pop_back();
        }
    }
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(k, n, 1, 0);
        return res;
    }
};

剪枝操作
①已选元素综合如果可以大于n，可以直接return(剪掉)。
②集合中所剩元素个数小于所需的元素个数。

class Solution {
private:
    vector<int> path;
    vector<vector<int>> res;
    void backtracking(int k, int n, int startIndex, int sum){
	    if(sum > n){
		    return;
	    }
        if(path.size() == k){
            if(sum == n){
                res.push_back(path);
                return;
            }
        }
        for(int i = startIndex; i <= 9 - (k - path.size()) + 1; ++i){
            sum += i;
            path.push_back(i); 
            backtracking(k, n, i + 1, sum);
            sum -= i;
            path.pop_back();
        }
    }
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(k, n, 1, 0);
        return res;
    }
};

17.电话号码的字母组合

题目： 给定一个仅包含数字 2-9 的字符串，返回所有它能表示的字母组合。答案可以按任意顺序返回。
给出数字到字母的映射如下（与电话按键相同）。注意 1 不对应任何字母。
链接： https://leetcode.cn/problems/letter-combinations-of-a-phone-number
思路：

vector<string> res存放结果组合
string path存放结果
字母和数字映射

const string letterMap[10] = {
"", //0
"", //1
"abc", //2
"def", //3
"ghi", //4
"jkl", //5
"mno", //6
"pqrs", //7
"tuv", //8
"wxyz", //9
};

回溯三部曲：
①传入参数
const string& digits, int index
index记录遍历第几个数字了，用来遍历digits的，同时index也表示树的深度。
②终止条件
index = digits.size()
③单层遍历逻辑

int digit = digits[index] - '0';//将index指向的数字转为int
string letters = letterMap[digit];//取数字对应的字符集
for(int i = 0; i < letters.size(); i++){
    s.push_back(letters[i]);//处理
    backtracking(digits, index + 1);//递归
    s.pop_back();//回溯
}

class Solution {
private:
    const string letterMap[10] = {
    "", //0
    "", //1
    "abc", //2
    "def", //3
    "ghi", //4
    "jkl", //5
    "mno", //6
    "pqrs", //7
    "tuv", //8
    "wxyz", //9
    };
public:
    vector<string> res;
    string path;
    void backtracking(const string& digits, int index){
        if(index == digits.size()){
            res.push_back(path);
            return;
        }
        int digit = digits[index] - '0';
        string letters = letterMap[digit];
        for(int i = 0; i < letters.size(); ++i){
            path.push_back(letters[i]);
            backtracking(digits, index+1);
            path.pop_back();
        }
    }
    vector<string> letterCombinations(string digits) {
        if(digits.size() == 0){
            return res;
        }
        backtracking(digits, 0);
        return res;
    }
};