LeetCode(47):全排列 II

Medium!

题目描述:

给定一个可包含重复数字的序列,返回所有不重复的全排列。

示例:

输入: [1,1,2]
输出:
[
  [1,1,2],
  [1,2,1],
  [2,1,1]
]

解题思路:

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

C++解法一:

 1 class Solution {
 2 public:
 3     vector<vector<int> > permuteUnique(vector<int> &num) {
 4         vector<vector<int> > res;
 5         vector<int> out;
 6         vector<int> visited(num.size(), 0);
 7         sort(num.begin(), num.end());
 8         permuteUniqueDFS(num, 0, visited, out, res);
 9         return res;
10     }
11     void permuteUniqueDFS(vector<int> &num, int level, vector<int> &visited, vector<int> &out, vector<vector<int> > &res) {
12         if (level >= num.size()) res.push_back(out);
13         else {
14             for (int i = 0; i < num.size(); ++i) {
15                 if (visited[i] == 0) {
16                     if (i > 0 && num[i] == num[i - 1] && visited[i - 1] == 0) continue;
17                     visited[i] = 1;
18                     out.push_back(num[i]);
19                     permuteUniqueDFS(num, level + 1, visited, out, res);
20                     out.pop_back();
21                     visited[i] = 0;
22                 }
23             }
24         }
25     }
26 };

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

C++解法二:

 1 class Solution {
 2 public:
 3     vector<vector<int>> permuteUnique(vector<int>& nums) {
 4         set<vector<int>> res;
 5         permute(nums, 0, res);
 6         return vector<vector<int>> (res.begin(), res.end());
 7     }
 8     void permute(vector<int> &nums, int start, set<vector<int>> &res) {
 9         if (start >= nums.size()) res.insert(nums);
10         for (int i = start; i < nums.size(); ++i) {
11             if (i != start && nums[i] == nums[start]) continue;
12             swap(nums[i], nums[start]);
13             permute(nums, start + 1, res);
14             swap(nums[i], nums[start]);
15         }
16     }
17 };

猜你喜欢

转载自www.cnblogs.com/ariel-dreamland/p/9143202.html