_22LeetCode代码随想录算法训练营第二十二天-回溯算法 | 216.组合总和III、17.电话号码的字母组合
题目列表
- 216.组合总和III
- 17.电话号码的字母组合
216.组合总和III
代码随想录地址:
https://programmercarl.com/0216.%E7%BB%84%E5%90%88%E6%80%BB%E5%92%8CIII.html
题目
找出所有相加之和为 n
的 k
个数的组合,且满足下列条件:
- 只使用数字1到9
- 每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
示例 1:
输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。
示例 2:
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
解释:
1 + 2 + 6 = 9
1 + 3 + 5 = 9
2 + 3 + 4 = 9
没有其他符合的组合了。
示例 3:
输入: k = 4, n = 1
输出: []
解释: 不存在有效的组合。
在[1,9]范围内使用4个不同的数字,我们可以得到的最小和是1+2+3+4 = 10,因为10 > 1,没有有效的组合。
提示:
2 <= k <= 9
1 <= n <= 60
思路
本题k相当于树的深度,9(因为整个集合就是9个数)就是树的宽度。
代码
/*
* @lc app=leetcode.cn id=216 lang=cpp
*
* [216] 组合总和 III
*/
// @lc code=start
class Solution {
private:
vector<vector<int>> res;
vector<int> data;
void backtracking(int k, int n, int sum, int start)
{
//剪枝,如果sum已经大于n了,那么这个组合已经不满足条件了,可以直接剪枝
if(sum > n)
return;
//终止条件
if(data.size() == k)
{
if(n == sum)
res.push_back(data);
return;
}
//这里剪枝了的
for(int i = start; i <= 9 - (k - data.size()) + 1; i++)
{
//让当前元素入栈
data.push_back(i);
sum += i;//将当前元素加到sum中
backtracking(k, n, sum, i + 1);//继续回溯
sum -= i;//回溯
data.pop_back();//回溯
}
}
public:
vector<vector<int>> combinationSum3(int k, int n) {
res.clear();
data.clear();
backtracking(k, n, 0, 1);
return res;
}
};
// @lc code=end
17.电话号码的字母组合
代码随想录地址:
https://programmercarl.com/0017.%E7%94%B5%E8%AF%9D%E5%8F%B7%E7%A0%81%E7%9A%84%E5%AD%97%E6%AF%8D%E7%BB%84%E5%90%88.html
题目
给定一个仅包含数字 2-9
的字符串,返回所有它能表示的字母组合。答案可以按 任意顺序 返回。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
示例 1:
输入:digits = "23"
输出:["ad","ae","af","bd","be","bf","cd","ce","cf"]
示例 2:
输入:digits = ""
输出:[]
示例 3:
输入:digits = "2"
输出:["a","b","c"]
提示:
0 <= digits.length <= 4
digits[i]
是范围['2', '9']
的一个数字。
思路
代码
/*
* @lc app=leetcode.cn id=17 lang=cpp
*
* [17] 电话号码的字母组合
*/
// @lc code=start
class Solution {
private:
const string letters[10] = {"", "", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz"};
vector<string> res;
string data;
void backtracking(string digits, int index)
{
//如果data的大小与digits的大小相等,则可以将data插入res中
if(data.size() == digits.size())
{
res.push_back(data);
return;
}
//获取键值
int letter = digits[index] - '0';
//将键值转换为字符串
string key = letters[letter];
//处理字符串
for(int i = 0; i < key.size(); i++)
{
data.push_back(key[i]);
backtracking(digits, index + 1);
data.pop_back();
}
}
public:
vector<string> letterCombinations(string digits) {
res.clear();
if(digits.size() == 0)
return res;
backtracking(digits, 0);
return res;
}
};
// @lc code=end