47. 全排列 II
这一题跟46题——全排列,非常类似。但是不同的是,这一题给的集合中有重复的元素。所以关键是要怎么处理这些重复的元素,在保证一个结果中可以出现重复数字的情况下,最后的结果集中又不能有一样的解。
举个例子,比如:nums = [1, 1’, 3], target = 5; 那么正确的结果为:
[
[1,1’,3],
[1,3,1’]
[3,1,1’]
]
可以看到,一个正确的解中可以包含重复的数字(1和1’),但结果中没有重复的解,比如[1, 1’, 3]和[1’, 1, 3]。
1. 如何避免重复?——还是排序!
这也是这道题目最关键的地方。我是做完了第40题-组合总和2再来做这道题的,所以很容易想到还是需要——排序,来去除重复元素。
我开始想到了,for循环中增加判断的条件:**nums[i]和nums[i -1]**是否相等( nums[i] == nums[i - 1])肯定仍然需要;但是因为是全排列,所以每次循环肯定都是从下标0开始的;同时,为了不使用之前使用过的数字,需要used[]数组来进行标记。总的来说,这题像是46和40的结合,既要避免重复,也要避免使用用过的数字。因此,刚开始我写的代码如下:
for (int i = 0; i < nums.size(); i++) {
if (i > 0 && nums[i] == nums[i - 1]) { //避免重复
continue;
}
if (used[i] == true) { // 跟46的写法一样,用过的数字不再用了。
continue;
}
...
经过验证,仅凭这个条件是不对的。上述代码的输出结果,为空[]。
分析之后发现,在求解单个path的时候,会错误的将部分结果舍去。具体的是:
(1) 在第一次执行backtrack函数时,当i = 0时,会像path中弹入nums[0],并将used[0]标记为true;此时path=[1’],然后会继续递归调用回溯函数backtrack();
(2) 来到避免重复的if判断语句,这个时候一判断,i = 0, 条件不成立,执行下一个if语句;
(3) 此时used[0]为真,执行continue语句,进入下一次循环;
(4) 此时i’的值变为1 > 0, 同时nums[1]=nums[0]=1,因此第一个if语句判断为真,跳过当前循环;
(5) i’接着自增为2,第一个if语句判断结果为false(nums[2]=3 != nums[1]=1’);第二个if语句也为真(used[2]=false);
(6)调用第三层递归,i’’‘从0开始,nums[0],nums[1]和nums[3]都跳过了(分别因为used[0]=true,nums[1]=nums[0]=1和nums[3]=true);
(7)i’再次递增,循环结束,没找到path,返回。
(8)i递增为1,然继续重复上面的过程。
从上面的过程可以看到,仅凭第一个if语句里面的两个条件是不够的,还需要增加条件使得在找一个path的过程中,选了1以后,还能选择1’。但是,再找下个path的时候,不能一开始选1’,即选了1’就不能选1。经过分析,新增了一个条件:use[i - 1] == false;
for (int i = d; i < nums.size(); i++) {
if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {// 为了避免重复
continue;
}
if (used[i] == true) {// 为了避免重复
continue;
}
...
这样一来,在上诉第(4)步骤中,虽然i’=1>0 且nums[1]=nums[0]=1,但是used[0]=true;使得if语句为假,会将nums[2],即1’加入path中!!但是,当i=1时,即我们第一个选择的数为1’时,在第一个if语句判断的时候,结果为真(i=1>0 && nums[1]=nums[0]=1 && used[0]=false),就会跳过当前循环执行下次循环,这样就避免了path=[1’, 1, 3]的情况!
理解:前面说过,for循环可以看为回溯的横向过程。增加了used[i - 1] == false这个条件,就可以避免第一个数选择1’的情况!
2. 完整的代码
void backtrack(vector<int>& nums, vector<vector<int>>& res, vector<bool> &used) {
if (path.size() == nums.size()) {
res.push_back(path);
return;
}
for (int i = 0; i < nums.size(); i++) {
if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {// 为了避免重复
continue;
}
if (used[i] == true) { // 跟不重复的写法一样,用过的数字不再用了。
continue;
}
path.push_back(nums[i]);
used[i] = true;
backtrack(nums, res, used);
path.pop_back();
used[i] = false;
}
}
总结
这题的关键是,怎么避免重复,关键是排序 + used[i - 1] == false。后面也算理解得比较清楚了。