题目:
输入一组数字(可能包含重复数字),输出其所有的排列方式。
样例:
输入:[1,2,3]
输出:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]
方法:(回溯法)
时间复杂度 O(n!)
1.先将所有数从小到大排序,使得相同的数排在一起;
2.从左到右依次枚举每个数,每次将它放在一个空位;
对于相同数,我们人为定序,避免重复计算
3.在dfs时记录一个额外的状态,记录上一个相同数存放的位置 start,我们在枚举当前数时,只枚举 start+1,start+2,…,nstart+1,start+2,…,n 这些位置,保证重复数字的相对位置不变。
4.在递归前和回溯时,对状态state进行更新。
表示状态state时,state是用二进制数表示对应位置是否已经插入元素,通过位操作实现判断和更新。
class Solution {
vector<vector<int>> ans;
vector<int> path;
public:
vector<vector<int>> permutation(vector<int>& nums) {
path.resize(nums.size());
sort(nums.begin(),nums.end());//先从小到大排序,使得重复元素靠在一起
dfs(nums,0,0,0);
return ans;
}
void dfs(vector<int>& nums, int u, int start, int state)
{
if(u == nums.size()){
ans.push_back(path);
return;
}
if(!u || nums[u] != nums[u-1]) start = 0;//当前枚举的是第一个元素,或当前枚举的值与前一个不同,此时start更新为0,表示可以从下标为0的位置开始放
for(int i = start; i < nums.size(); i++){
if(!(state >> i & 1)){//第i位是否已经插入元素
path[i] = nums[u];//将该元素加入结果path
dfs(nums, u+1, i+1, state + (1<<i));
}
}
}
};