首页 > 代码库 > [C++]LeetCode: 120 Permutations II

[C++]LeetCode: 120 Permutations II

题目:

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的区别就是,输入的数字数组中包含重复的元素。如果我们对重复的元素不做处理的话,我们就会产生重复的结果。那么如何避免这种重复呢?按照我们前面处理这类问题的方法,首先对输入数组进行排序,然后对于重复的元素循环时跳过递归函数的调用,只对第一个未被使用的元素进行递归,这一次结果会出现在第一个的递归函数结果中,因为后面的重复元素将重复此次的递归结果,后面的结果被略过。如果第一个重复元素前面的元素还没在前面的结果(本次tmp)中(!used[i-1]), 那么我们不需要递归。比如[1,1,1,2], 如果第一个1在结果中(used[0] = true),我们下次循环时,我们判断到第二个1,才进行递归和push进tmp, 否则会导致重复。这一点很重要,需要思考下。总体来说,我们需要对重复元素和前面元素的使用情况进行判断。

AC Code:

class Solution {
public:
    vector<vector<int> > permuteUnique(vector<int> &num) {
        vector<vector<int> > ret;
        if(num.size() == 0) return ret;
        
        vector<int> tmp;
        vector<bool> used(num.size(), false);
        sort(num.begin(), num.end());
        permuteUnique_helper(num, tmp, used, ret);
        return ret;
    }

private:
    void permuteUnique_helper(vector<int>& num, vector<int> tmp, vector<bool> used, vector<vector<int> >& ret)
    {
        if(tmp.size() == num.size())
        {
            ret.push_back(tmp);
            return;
        }
        
        for(int i = 0; i < num.size(); i++)
        {
            if(i > 0 && !used[i-1] && num[i-1] == num[i]) continue;
            
            if(!used[i])
            {
                tmp.push_back(num[i]);
                used[i] = true;
                permuteUnique_helper(num, tmp, used, ret);
                tmp.pop_back();
                used[i] = false;
            }
        }
        return;
    }
};
这个解法带有一般性,放到Permutations中也是正确的,所以在面试时遇到,我们可以直接实现这个代码,不要假设元素没有重复,可以和面试官进行讨论,但是一般我们都要考虑重复元素。


这道题,我们也可以用迭代的方法去做,可以用next permutation的算法,保证顺序,从最小的排列到最大的排列迭代下去,代码相对冗长,一般面试不会考到这道题迭代的方法,不过可以看下这篇博文:

Permutations, Permutations II(求全排列)




[C++]LeetCode: 120 Permutations II