- leetcode: Permutations II | LeetCode OJ
- lintcode: (16) Permutations II
Given a list of numbers with duplicate number in it. Find all unique permutations.
For numbers [1,2,2]
the unique permutations are:
[
[1,2,2],
[2,1,2],
[2,2,1]
]
Do it without recursion.
在上题的基础上进行剪枝,剪枝的过程和 Unique Subsets 一题极为相似。为了便于分析,我们可以先分析简单的例子,以
$$[1, 2_1, 2_2]$$ $$[1, 2_2, 2_1]$$ $$[2_1, 1, 2_2]$$ $$[2_1, 2_2, 1]$$ $$[2_2, 1, 2_1]$$ $$[2_2, 2_1, 1]$$
从以上结果我们注意到1
和2
重复,5
和3
重复,6
和4
重复,从重复的解我们可以发现其共同特征均是第二个
首先可以确定 list
为 list.pop_back()
之后,list
为 list
的将为 list
中,如何才能做到这一点呢?我们仍然从上述例子出发进行分析,在第一次加入 visited[1]
为true
(对应 visited[1]
为false
,因为在list
为 list.pop_back()
后即置为false
。
一句话总结即为:在遇到当前元素和前一个元素相等时,如果前一个元素visited[i - 1] == false
, 那么我们就跳过当前元素并进入下一次循环,这就是剪枝的关键所在。另一点需要特别注意的是这种剪枝的方法能使用的前提是提供的nums
是有序数组,否则无效。
class Solution {
public:
/**
* @param nums: A list of integers.
* @return: A list of unique permutations.
*/
vector<vector<int> > permuteUnique(vector<int> &nums) {
vector<vector<int> > ret;
if (nums.empty()) {
return ret;
}
// important! sort before call `backTrack`
sort(nums.begin(), nums.end());
vector<bool> visited(nums.size(), false);
vector<int> list;
backTrack(ret, list, visited, nums);
return ret;
}
private:
void backTrack(vector<vector<int> > &result, vector<int> &list, \
vector<bool> &visited, vector<int> &nums) {
if (list.size() == nums.size()) {
result.push_back(list);
// avoid unnecessary call for `for loop`, but not essential
return;
}
for (int i = 0; i != nums.size(); ++i) {
if (visited[i] || (i != 0 && nums[i] == nums[i - 1] \
&& !visited[i - 1])) {
continue;
}
visited[i] = true;
list.push_back(nums[i]);
backTrack(result, list, visited, nums);
list.pop_back();
visited[i] = false;
}
}
};
Unique Subsets 和 Unique Permutations 的源码模板非常经典!建议仔细研读并体会其中奥义。
后记:在理解 Unique Subsets 和 Unique Permutations 的模板我花了差不多一整天时间才基本理解透彻,建议在想不清楚某些问题时先分析简单的问题,在纸上一步一步分析直至理解完全。
Permutation 的题使用字典序的做法其实更为简单,且为迭代的解法,效率也更高。代码和之前的 Permutations 那道题一模一样。
class Solution {
/**
* @param nums: A list of integers.
* @return: A list of unique permutations.
*/
public ArrayList<ArrayList<Integer>> permuteUnique(ArrayList<Integer> nums) {
ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
if (nums == null || nums.size() == 0) return result;
// deep copy(do not change nums)
List<Integer> perm = new ArrayList<Integer>(nums);
// sort first!!!
Collections.sort(perm);
while (true) {
// step0: add perm into result
result.add(new ArrayList<Integer>(perm));
// step1: search the first num[k] < num[k+1] backward
int k = -1;
for (int i = perm.size() - 2; i >= 0; i--) {
if (perm.get(i) < perm.get(i + 1)) {
k = i;
break;
}
}
// if current rank is the largest, exit while loop
if (k == -1) break;
// step2: search the first perm[k] < perm[l] backward
int l = perm.size() - 1;
while (l > k && perm.get(l) <= perm.get(k)) l--;
// step3: swap perm[k] with perm[l]
Collections.swap(perm, k, l);
// step4: reverse between k+1 and perm.length-1;
reverse(perm, k + 1, perm.size() - 1);
}
return result;
}
private void reverse(List<Integer> nums, int lb, int ub) {
for (int i = lb, j = ub; i < j; i++, j--) {
Collections.swap(nums, i, j);
}
}
}
见前一题,略。
略