这题是一个回溯的思想,类似于DFS。这里举一个例子来说明。
我们有一个数组,里面有三个数字,分别是1、2、3,我们先以1为开头,得到的全部排列为:
1、2、3
和1、3、2
同理,我们再分别以2和3作为开头,得到的全排列分别为:
2、1、3
和2、3、1
3、1、2
和3、2、1
由上述的例子我们可以发现,以一个数为开头的全排列的数量就是其之后所有数的全排列的方式的和,这就是我们如何去实现的理论依据。
我们设置一个变量index
,用以描述我们当前遍历的层数,当层数等于源数组长度时,说明此时已出现一个答案,推进答案并返回上一层状态,对于每一个数,都依次和后续数字进行位置交换,这样就可以得到所有的结果,每一次退回上层时重置状态即可。思路有了,AC代码如下:
class Solution {
private:
void backtrack(vector<int>& nums, int index, vector<vector<int>>& ans) {
if(index == nums.size()-1) {
ans.push_back(nums);
return;
}
for(int i=index;i<=nums.size()-1;i++) {
swap(nums[i], nums[index]);
backtrack(nums, index+1, ans);
swap(nums[i], nums[index]);
}
}
public:
vector<vector<int>> permute(vector<int>& nums) {
vector<vector<int>> ans;
backtrack(nums, 0, ans);
return ans;
}
};