Leetcode算法入门第十一天(递归 / 回溯)
77. 组合
题目描述
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
样例
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
输入:n = 1, k = 1
输出:[[1]]
思路
回溯法:有回溯就有递归。
参考代码
class Solution {
public:
vector<vector<int>> res;//满足条件的所有组合的集合
vector<int> path;//符合条件的单个组合
//用回溯法
void backtracking(int n,int k,int index)
{
//递归终点:当单个组合path中元素满足k个,就可以终止
if(path.size()==k)
{
res.push_back(path);
return;
}
//处理单层递归逻辑
//开始遍历递归树:
for(int i=index;i<=n;i++)
{
//存入当前下标所指的元素,确定这一组合的第一个元素
path.push_back(i);
//调用自身:缩小搜索规模,寻找
backtracking(n,k,i+1);
//回溯:
path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backtracking(n,k,1);
return res;
}
};
46. 全排列
题目描述
给定一个不含重复数字的数组 nums ,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。
样例
输入:nums = [1,2,3]
输出:[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
输入:nums = [0,1]
输出:[[0,1],[1,0]]
思路
回溯法
参考代码
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
//used数组:标志已经保存过的元素
void backtracking(vector<int>& nums,vector<bool>& used)
{
//递归终点:满足条件的排列
if(path.size()==nums.size())
{
//存放当前排列
res.push_back(path);
return;
}
for(int i=0;i<nums.size();i++)
{
if(used[i]==true) continue;
//保存当前节点
path.push_back(nums[i]);
used[i]=true;
//递归调用
backtracking(nums,used);
//回溯:
//弹出刚才压入的元素,并调整标志
path.pop_back();
used[i]=false;
}
}
vector<vector<int>> permute(vector<int>& nums) {
vector<bool> used(nums.size(),false);
backtracking(nums,used);
return res;
}
};
784. 字母大小写全排列
题目描述
给定一个字符串 s ,通过将字符串 s 中的每个字母转变大小写,我们可以获得一个新的字符串。
返回 所有可能得到的字符串集合 。以 任意顺序 返回输出。
样例
输入:s = "a1b2"
输出:["a1b2", "a1B2", "A1b2", "A1B2"]
输入: s = "3z4"
输出: ["3z4","3Z4"]
思路
回溯法
参考代码
class Solution {
public:
vector<string> res;//所有满足的字符串集合
void backtracking(string&s,int index)
{
//如果字母的下标
if(index==s.size())
{
res.push_back(s);
return;
}
//如果是数字,就跳过当前的字符,递归:调整下一个字母
if(s[index]>='0'&&s[index]<='9')
{
backtracking(s,index+1);
}
else//是字母
{
//把当前字母改成小写
s[index]=tolower(s[index]);
backtracking(s,index+1);
//回溯:
//把当前字母改成大写
s[index]=toupper(s[index]);
backtracking(s,index+1);
}
}
vector<string> letterCasePermutation(string s) {
backtracking(s,0);
return res;
}
};