在全排列问题上加上一个剪枝就是如果当前的arr(i)==arr(i-1)(解空间树的同一层)同时visited[i-1]没有被访问过,说明就重复了。 注意初始数组一定要有序,否则需要进行排序
class Solution {
public:
vector<vector<int> > res;
vector<int> path;
int visited[10];
void dfs_back(vector<int> &nums,int index)
{
if(index==nums.size())
{
res.push_back(path);
return;
}
else
{
for(int i=0;i<nums.size();i++)
{
if(visited[i]||(i>0&&nums[i]==nums[i-1]&&visited[i-1])) continue;
visited[i] = 1;
path.push_back(nums[i]);
dfs_back(nums,index+1);
visited[i]=0;
path.pop_back();
}
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
sort(nums.begin(),nums.end());
dfs_back(nums,0);
return res;
}
};
|