当有重复元素的时候呢?
不用拍脑袋都会想到一种方法,也是所有有重复元素时的通用处理方法,维护一个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;
}
};