感觉这几个回溯的题目思路都蛮简单,放一个简单模板
if(得到一个解)
{
存入答案
返回
}
if(无法得到解)
{
返回
}
if(数据可行)
{
存入数据
递归执行下一层
弹出刚存入的数据
递归执行下一层
}
思路
从第一个位置开始放,对于每个数,我们可以选择放入或者不放入,当放的数量到k的时候,就保留下结果。
class Solution {
public:
vector<vector<int>> ans;
vector<int> temp;
void bfs(int n,int k,int m)
{
//数组中和接下来的数不够k时,不能拼成,返回
if(temp.size()+(n-m+1)<k)
{
return;
}
//将拼成的结果存入答案中
if(temp.size()==k)
{
ans.push_back(temp);
return;
}
//存入数据
temp.push_back(m);
bfs(n,k,m+1);
//弹出刚存入的数据
temp.pop_back();
bfs(n,k,m+1);
}
vector<vector<int>> combine(int n, int k) {
bfs(n,k,1);
return ans;
}
};
思路
可惜这个题是vector类型的数组,不能用next_permutation,不然直接小到大排序在next_permutation就完事了。
开始递归回溯的思路,还是从第一个位置开始放,每个位置放的元素可以是nums中的任意一个元素,但是同一个数组方案中不能存在相同的元素。因此我用了一个无序表unordered_set来去重,(官方用了一个更优秀的方法来降低空间复杂度,等下放在后面)。当放到第nums.size()个元素时,可以将这个方案存入答案中。
class Solution {
public:
vector<vector<int>> ans;
vector<int> temp;
//去重
unordered_set<int> book;
void dfs(int cur,vector<int>&nums)
{
//放完了
if(cur==nums.size())
{
ans.push_back(temp);
}
for(int i=0;i<nums.size();i++)
{
//count()统计book表中nums[i]的数量
//保证表中无nums[i]
if(!book.count(nums[i]))
{
//book中加入nums[i],并将其存入数组中
book.insert(nums[i]);
temp.push_back(nums[i]);
dfs(cur+1,nums);
//把刚存入的元素再弹出(这里用的是循环,看下个数能不能存,因此不调用递归)
book.erase(nums[i]);
temp.pop_back();
}
}
}
vector<vector<int>> permute(vector<int>& nums) {
dfs(0,nums);
return ans;
}
};
class Solution {
public:
void backtrack(vector<vector<int>>& res, vector<int>& output, int first, int len){
// 所有数都填完了
if (first == len) {
res.emplace_back(output);
return;
}
for (int i = first; i < len; ++i) {
// 动态维护数组
swap(output[i], output[first]);
// 继续递归填下一个数
backtrack(res, output, first + 1, len);
// 撤销操作
swap(output[i], output[first]);
}
}
vector<vector<int>> permute(vector<int>& nums) {
vector<vector<int> > res;
backtrack(res, nums, 0, (int)nums.size());
return res;
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/permutations/solution/quan-pai-lie-by-leetcode-solution-2/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
思路
对于每个字母的,有两种选择,可以将他改为大写,也可以改为小写
然后判断下一个字母,如果是数字的话,则直接判断下一个字母
class Solution {
public:
vector<string> ans;
void bfs(string &s,int cur)
{
//执行完毕,插入结果
if(cur==s.length())
{
ans.push_back(s);
}
//处理数字
if(s[cur]>='0'&&s[cur]<='9')
{
bfs(s,cur+1);
}
//处理字母
if((s[cur]>='A'&&s[cur]<='Z')||(s[cur]>='a'&&s[cur]<='z'))
{
s[cur]=tolower(s[cur]);
bfs(s,cur+1);
s[cur]=toupper(s[cur]);
bfs(s,cur+1);
}
}
vector<string> letterCasePermutation(string s) {
bfs(s,0);
return ans;
}
};