【LeetCode】 77. 组合
题意:给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
思路:实际上是遍历一棵多叉树,树的结点是取值,树的路径就是答案。利用回溯,为了避免重复,每次从一个startidx开始取数,当满足k个数时返回。
代码:
class Solution {
public:
vector<vector<int> > m_res;
vector<int> m_path;
void backtracking(int n, int k, int startIdx)
{
if (k == m_path.size())
{
m_res.push_back(m_path);
return;
}
for (int i = startIdx; i <= n; ++i)
{
m_path.push_back(i);
backtracking(n, k, i + 1);
m_path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backtracking(n, k, 1);
return m_res;
}
};
【LeetCode】 216.组合总和III
题意:找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:
只使用数字1到9
每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
思路:和为n的k个数,也就是一棵多叉树,高为k,路径满足条件和为n
代码:
class Solution {
public:
vector<int> m_path;
vector<vector<int>> m_res;
void backtracking(int k, int n, int startIdx, int num)
{
if (k == m_path.size())
{
if (n == num)
{
m_res.push_back(m_path);
}
return;
}
for (int i = startIdx; i <= 9; ++i)
{
m_path.push_back(i);
backtracking(k, n, i + 1, num + i);
m_path.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(k, n, 1, 0);
return m_res;
}
};
【LeetCode】 17.电话号码的字母组合
题意:给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。答案可以按 任意顺序 返回。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
思路:输入字符串的长度就是多叉树的高度,每个字符对应到字母的映射就是每一层结点的取值范围。用哈希表存储数字到字符串的映射,根据输入同时遍历多个字符串,记录输入数字的下标。
代码:
class Solution {
public:
vector<string> m_res;
string m_path;
unordered_map<char, string> m_mp;
void backtracking(string digits, int idx)
{
if (m_path.size() == digits.size())
{
m_res.push_back(m_path);
return;
}
for (int i = idx; i < digits.size(); ++i)
{
char num = digits[i];
for (int j = 0; j < m_mp[num].size(); ++j)
{
m_path.push_back(m_mp[num][j]);
backtracking(digits, i + 1);
m_path.pop_back();
}
}
}
vector<string> letterCombinations(string digits) {
if (digits.empty())
{
return m_res;
}
m_mp['2'] = "abc";
m_mp['3'] = "def";
m_mp['4'] = "ghi";
m_mp['5'] = "jkl";
m_mp['6'] = "mno";
m_mp['7'] = "pqrs";
m_mp['8'] = "tuv";
m_mp['9'] = "wxyz";
backtracking(digits, 0);
return m_res;
}
};
心态:“第七章 回溯算法part01” 拿下!
参考资料