Given a collection of distinct integers, return all possible permutations.
Example:
Input: [1,2,3]
Output:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]
看了看大佬们的讲解,终于自己做出来一道回溯的题。
回溯大致分为三步:
- 终止条件:回溯到什么情况就可以加入答案列表
- 剪枝条件:什么情况下就可以停止回溯直接continue
- 回溯范围:如何将当前层和下一层联系起来继续递归下去
对于这道全排列的题,我们需要一个res保存答案,一个temp参与回溯,当达到终于条件也就是temp的大小与nums的相等时便把temp加入res中;
然后需要一个check数组来标记使用过的元素,当一个数已经被使用过便把对应的位置的值设为1,这样避免了一个数重复使用多次。
由于这道题是要列出所有的排列结果,所以需要全部遍历
代码如下:
class Solution {
public:
vector<vector<int>> res;
void dfs(vector<int>& nums,vector<int>& temp,vector<int>& check){
if(temp.size()==nums.size()){
res.push_back(temp);
return;
}
for(int i=0;i<nums.size();i++){
if(check[i])
continue;
check[i]=1;
temp.push_back(nums[i]);
dfs(nums,temp,check);
temp.pop_back();
check[i]=0;
}
}
vector<vector<int>> permute(vector<int>& nums) {
vector<int> check(nums.size(),0);
vector<int> temp;
dfs(nums,temp,check);
return res;
}
};