當有重複元素的時候呢?
不用拍腦袋都會想到一種方法,也是所有有重複元素時的通用處理方法,維護一個set,如果這個元素沒加入過就加入,加入過了的忽略掉。但是,在這道題上這個通用方法居然逾時了!
怎麼辦?想一下為什麼會這樣,假設我們要排列的數字是1111112,噹噹前的排列中沒有1時,取哪個1產生一遍,都是一樣的。只有當前面的1都用過了,必須輪到這個1出場的時候,它才會有價值。更明確一點說,假設我們要在產生的排列中放兩個1,那麼這兩個1是原來的哪兩個根本無所謂,不斷的選,最終的結果肯定一樣,但是當我們要在排列中放3個1的時候,再選擇的1一定是新的了,是有意義的。
用演算法的語言描述就是,先把所有的候選數字排一下序,相同的數字會挨在一起,對於同樣的數字,都先取第一個加入排列,後面的相同的數字想加入排列,必須保證它前面的相同數字已經在排列中了,這樣避免了不斷產生重複的排列。
class Solution {public: set<vector<int> > vis; void getPerm(vector<int> &num, vector<int> &tpres, vector<vector<int> > &res, int len, int n, bool *visited){ if(len == n){ if(vis.find(tpres) == vis.end()){ vis.insert(tpres); res.push_back(tpres); } return; } for(int i=0;i<n;i++){ if(!visited[i]){ if(i!=0&&num[i] == num[i-1]&&!visited[i-1]) continue; visited[i] = 1; tpres.push_back(num[i]); getPerm(num, tpres, res, len+1, n, visited); visited[i] = 0; tpres.pop_back(); } } } vector<vector<int> > permuteUnique(vector<int> &num) { vector<vector<int> > res; vector<int> tpres; int msize = num.size(); if(msize<=0) return res; bool visited[msize]; memset(visited, 0, sizeof(visited)); sort(num.begin(), num.end()); getPerm(num, tpres, res, 0, msize, visited); return res; }};