LeetCode 47全排列II
-
题目简述:给定一个可包含重复数字的序列,返回所有不重复的全排列。
-
输入:[1, 1, 2]
输出:[ [1,1,2], [1,2,1], [2,1,1]]
-
思路:DFS + 回溯 + 剪枝
在一定会产生重复结果集的地方剪枝。产生重复结点的地方,正是图中标注了“剪刀”,且被绿色框框住的地方。
搜索过程只要遇到起点一样,就有可能产生重复。
- 在图中 ② 处,搜索的数也和上一次一样,但是上一次的
1
还在使用中; - 在图中 ① 处,搜索的数也和上一次一样,但是上一次的
1
刚刚被撤销,正是因为刚被撤销,下面的搜索中还会使用到,因此会产生重复,剪掉的就应该是这样的分支。
class Solution {
public:
vector<vector<int>> ans;
vector<bool> st;
vector<int> path;
vector<vector<int>> permuteUnique(vector<int>& nums) {
sort(nums.begin(), nums.end());//排序为了方便剪枝
st = vector<bool>(nums.size(), false);
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])
{
//剪枝条件,i>0保证nums[i-1]有意义
//st[i-1] == false是因为nums[i-1]在回退的过程中刚刚被撤销选择
//表示如果当前数等于上一个数并且上一个数刚刚被撤销选择,那么不选当前数
if(i > 0 && nums[i - 1] == nums[i] && st[i - 1] == false)
continue;
st[i] = true;
path.push_back(nums[i]);
dfs(nums, u + 1);
st[i] = false;
path.pop_back();
}
}
};