一、前言
参考文献:代码随想录
今天的还是回溯算法,主要用到了昨天的回溯组合方法和巧妙思路方法,让我们继续为算法打基础吧!
二、组合总和|||
1、思路:
这一题和昨日的组合没啥太大区别只是遍历的范围变为了固定的1~9,所以只需要在for循环中做做手脚即可。
(1)首先还是确定返回值和参数,这里和昨天的一样,返回void,参数三个参数,组合大小,总和大小,以及开始位置
void backtracking(int k, int n, int Instart)
(2)接着就是终止条件了,这个需要判断两个条件,以来确认是否符合要求
if (sum == n && path.size() == k) {result.push_back(path);return;}
长度和大小要符合要求。
(3)接着就是for循环和递归回溯了,最开始的版本是
for (int i = Instart; i < 10; i++) {path.push_back(i);sum += i;backtracking(k ,n, i + 1);// 回溯,重新利用path.pop_back();sum -= i;
这里就没有涉及到剪枝的操作,而只是全部暴力递归了,然而我们发现sum>n时,就可以提前回溯了,因为后面的肯定不符合要求了。
for (int i = Instart; i < 10 - (k - path.size()) + 1; i++) {path.push_back(i);sum += i;// 剪枝操作,只要sum大于n就可以直接剪掉了if (sum > n) {path.pop_back();sum -= i;return;}backtracking(k ,n, i + 1);// 回溯,重新利用path.pop_back();sum -= i;}
2、整体代码如下:
class Solution {
private:vector<int> path; // 存储单个组合vector<vector<int>> result; // 返回结果int sum; // 统计是否等于nvoid backtracking(int k, int n, int Instart) {if (sum == n && path.size() == k) {result.push_back(path);return;}for (int i = Instart; i < 10 - (k - path.size()) + 1; i++) {path.push_back(i);sum += i;// 剪枝操作,只要sum大于n就可以直接剪掉了if (sum > n) {path.pop_back();sum -= i;return;}backtracking(k ,n, i + 1);// 回溯,重新利用path.pop_back();sum -= i;}}
public:vector<vector<int>> combinationSum3(int k, int n) {backtracking(k, n, 1);return result;}
};
三、电话号码的字母组合
1、思路:
这个题目,我根据我的印象做了,但是出不来,主要是对c++的STL不熟悉所以导致出不来结果,然后只能请问GPT教授了:
(1)首先这个题目也是利用回溯,即递归里面套循环,这里我们的返回值和参数如下:
void backtracking(string digits, int startI, int startJ, vector<string> s)
其实这里面的startJ可以省略,因为他一直是从0开始的,从头开始遍历,找到组合项,digits就是需要按的按键,startI为按键的开始位置,s就是按键所对应的数字了
(2)终止条件,也很普遍,这里就不多说了
if (path.size() == digits.size()) {result.push_back(path);return;}
(3)接着就是循环和递归了,这里面略微有些技巧,但是不多
for (int i = startI; i < digits.size(); i++) {// 判断这个按键包含数字的多少int digit = digits[i] - '2';for (int j = startJ; j < s[digit].size(); j++) {path.push_back(s[digit][j]);backtracking(digits, i + 1, 0, s);path.pop_back();}}
一共嵌套两层循环,第一层是按键的位置,第二层是按键包含的字母,就开始递归,回溯了。
2、整体代码如下:
class Solution {
private:string path;vector<string> result;void backtracking(string digits, int startI, int startJ, vector<string> s) {if (path.size() == digits.size()) {result.push_back(path);return;}for (int i = startI; i < digits.size(); i++) {int digit = digits[i] - '2';for (int j = startJ; j < s[digit].size(); j++) {path.push_back(s[digit][j]);backtracking(digits, i + 1, 0, s);path.pop_back();}}}
public:vector<string> letterCombinations(string digits) {if (digits.size() == 0) {return result;}vector<string> s = {"abc","def","ghi","jkl","mno","pqrs","tuv","wxyz"};backtracking(digits, 0, 0, s);return result;}
};
今日学习时间:1.5小时
leave message:
The boundless forest sheds its leaves shower by shower; the endless river rolls its waves hour after hour.
无边落木萧萧下,不尽长江滚滚来。