LeetCode--Permutations II 全排列之二

2018-01-25 10:57:27来源:网络收集作者:管理员人点击

分享
第七城市

[var1]


Given a collection of numbers that might contain duplicates, return all possible unique permutations.



For example,
[1,1,2] have the following unique permutations:
[1,1,2], [1,2,1], and [2,1,1].



 



这道题是之前那道 Permutations
全排列的延伸,由于输入数组有可能出现重复数字,如果按照之前的算法运算,会有重复排列产生,我们要避免重复的产生,在递归函数中要判断前面一个数和当前的数是否相等,如果相等,前面的数必须已经使用了,即对应的visited中的值为1,当前的数字才能使用,否则需要跳过,这样就不会产生重复排列了。



 对于if (i > 0 && num[i] == num[i - 1] && visited[i - 1] == 0) continue;的理解:



假设有两个1,排序后位置不同,我们规定这两个1在排序中出现的顺序必须和数组中位置顺序一样,也就是第一个1只能出现在前面,第二个1只能出现在后面,这样就消除了重复解。
对应到代码中,要排除的情况就是在前面的位置选择第二个1,这时检查发现第一个1还没用过就是这种情况,于是可以跳过了。




解法一:


class Solution {
public:
vector > permuteUnique(vector &num) {
vector > res;
vector out;
vector visited(num.size(), 0);
sort(num.begin(), num.end());
permuteUniqueDFS(num, 0, visited, out, res);
return res;
}
void permuteUniqueDFS(vector &num, int level, vector &visited, vector &out, vector > &res) {
if (level >= num.size()) res.push_back(out);
else {
for (int i = 0; i < num.size(); ++i) {
if (visited[i] == 0) {
if (i > 0 && num[i] == num[i - 1] && visited[i - 1] == 0) continue;
visited[i] = 1;
out.push_back(num[i]);
permuteUniqueDFS(num, level + 1, visited, out, res);
out.pop_back();
visited[i] = 0;
}
}
}
}
};


还有一种比较简便的方法,在Permutations的基础上稍加修改,我们用set来保存结果,利用其不会有重复项的特点,然后我们再递归函数中的swap的地方,判断如果i和start不相同,但是nums[i]和nums[start]相同的情况下跳过,继续下一个循环,参见代码如下:



 



解法二:


class Solution {
public:
vector> permuteUnique(vector& nums) {
set> res;
permute(nums, 0, res);
return vector> (res.begin(), res.end());
}
void permute(vector &nums, int start, set> &res) {
if (start >= nums.size()) res.insert(nums);
for (int i = start; i < nums.size(); ++i) {
if (i != start && nums[i] == nums[start]) continue;
swap(nums[i], nums[start]);
permute(nums, start + 1, res);
swap(nums[i], nums[start]);
}
}
};

转自:http://www.cnblogs.com/grandyang/p/4359825.html



第七城市

最新文章

123

最新摄影

闪念基因

微信扫一扫

第七城市微信公众平台