回溯算法
回溯算法框架(重点)
解决一个回溯问题,实际上就是一个决策树的遍历过程。
站在回溯树的一个节点上,你只需要思考 3 个问题:
1、路径:也就是已经做出的选择。
2、选择列表:也就是你当前可以做的选择。
3、结束条件:也就是到达决策树底层,无法再做选择的条件。
回溯算法框架
result;
void backtracking(路径、选择列表) {
if (终止条件) {
result.add(存放结果);
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
排列问题
46. 全排列
给定一个不含重复数字的数组 nums
,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。
思路
画出回溯树
从前往后,一位一位枚举,每次选择一个没有被使用过的数。
选好之后,将该数的状态改成**“已被使用**”,同时将该数记录在相应位置上,然后递归。
递归返回时,不要忘记将该数的状态改成“未被使用”,并将该数从相应位置上删除。
代码实现:
class Solution {
public:
vector<vector<int>> ans;
vector<int> path;
vector<bool> st;
vector<vector<int>> permute(vector<int>& nums) {
path = vector<int>(nums.size());
st = vector<bool>(nums.size());
dfs(nums, 0);
return ans;
}
void dfs(vector<int>& nums, int u) {
if (u == nums.size()) {
ans.push_back(path);
return;
}
for (int i = 0; i < nums.size(); i ++ ) {
if (st[i] == false) {
path[u] =nums[i];
st[i] = true;
dfs(nums, u + 1);
st[i] = false;
}
}
}
};
47. 全排列 II
给定一个可包含重复数字的序列 nums
,按任意顺序 返回所有不重复的全排列。
思路:
这道题目和46.全排列的区别在与给定一个可包含重复数字的序列,要返回所有不重复的全排列。
解决办法:
本质就是对于重复的点之间相对位置不变
去重一定要对元素进行排序,这样我们才方便通过相邻的节点来判断是否重复使用了。
重点就是剪枝:如果这个数和之前的数一样,并且之前的数还未使用过(说明已经回溯过)
class Solution {
public:
vector<vector<int>> ans;
vector<int> path;
vector<bool> st;
vector<vector<int>> permuteUnique(vector<int>& nums) {
path = vector<int>(nums.size());
st = vector<bool>(nums.size());
sort(nums.begin(),nums.end());
dfs(nums,0);
return ans;
}
void dfs(vector<int>&nums,int u){
if(u == nums.size()){
ans.push_back(path);
return;
}
for(int i = 0;i < nums.size();i ++){
if(!st[i]){
if(i && nums[i - 1] == nums[i] && !st[i - 1]) continue;
st[i] = true;
path[u] = nums[i];
dfs(nums,u + 1);
st[i] = false;
}
}
}
};