[LeetCode] 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].

1 我的思路,在Permutations的基础上对结构去重,不过超时了。。

class Solution {
        vector<vector<int> > m_result;
    public:
        void swap(int & x, int & y)
        {
            int tmp = x;
            x = y;
            y = tmp;
        }
        void dfs(vector<int> & num, int dep )
        {
            if(dep == num.size())
            {
                m_result.push_back(num);
                return;
            }   

            for(int i = dep; i < num.size(); i++)
            {   
                    //if(i != dep && num[i] == num[dep])
                    //    continue;
                    swap(num[i], num[dep]);
                    dfs(num, dep + 1); 
                    swap(num[i], num[dep]);
            }   
        }   
    
        vector<vector<int> > permuteUnique(vector<int> &num)
        {   
            dfs( num, 0); 

            //erase the duplicate
            sort(m_result.begin(), m_result.end());
            m_result.erase(unique(m_result.begin(), m_result.end()), m_result.end());

            return m_result;
        }
};

思路2 copy from http://www.cnblogs.com/remlostime/archive/2012/11/13/2768816.html 

先对数组进行排序,这样在DFS的时候,可以先判断前面的一个数是否和自己相等,相等的时候则前面的数必须使用了,自己才能使用,这样就不会产生重复的排列了。

解释:假设有相同的3个2,排序后按照顺序称为2a,2b,2c, 由于保证前面数使用后,后面的数才能有,保证了最后的结果中2a在2b之前,2b在2c之前,这样就避免了重复。

 

class Solution {
private:
    bool canUse[100];
    int a[100];
    vector<vector<int> > ret;
public:
    void dfs(int dep, int maxDep, vector<int> &num)
    {
        if (dep == maxDep)
        {
            vector<int> ans;
            for(int i = 0; i < maxDep; i++)
                ans.push_back(a[i]);
            ret.push_back(ans);
            return;
        }
        
        for(int i = 0; i < maxDep; i++)
            if (canUse[i])
            {
                if (i != 0 && num[i] == num[i-1] && canUse[i-1])
                    continue;
                    
                canUse[i] = false;
                a[dep] = num[i];
                dfs(dep + 1, maxDep, num);
                canUse[i] = true;
            }
    }
    
    vector<vector<int> > permuteUnique(vector<int> &num) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        sort(num.begin(), num.end());
        memset(canUse, true, sizeof(canUse));
        ret.clear();
        dfs(0, num.size(), num);
        return ret;
    }
};

 

posted @ 2015-02-12 17:06  穆穆兔兔  阅读(214)  评论(0编辑  收藏  举报