回溯算法解决子集和排列问题有以下区别,该类问题中去重逻辑是最重要的环节,其他部分则为回溯问题的基本框架,比较好理解。本文主要解释子集和排列问题的去重逻辑对比。(组合是子集的一种特殊情况,就不再单一列举)
对于没有重复的子集和排列问题
子集问题
如: nums = [1, 2, 3]
其所有子集为:[ [],[1],[2],[3],[1,2],[1,3],[2,3],[1,2,3] ]
该问题中没有重复元素,不可以重复选择。
- 我们可以看到。首先该子集问题不限制子集的长度,所以我们在收集结果的时候不必检查其长度,任意长度的子集都应该被收集
- 该问题不可以回头寻找元素,例如:[1, 2]和[2, 1]是重复元素。所以其去重逻辑为:定义一个
start
变量来记录遍历开始的位置,而这个start
变量只能增加不能减小,也就意味着不可以回头寻找元素。
排列问题
如: nums = [1,2,3]
其所有的全排列为
[ [1,2,3],[1,3,2], [2,1,3],[2,3,1], [3,1,2],[3,2,1] ]
该问题中,没有重复元素,不可以重复选择。
- 首先,在收集结果的时候,我们需要判断结果的长度是否等于
nums.length
,只有等于的时候才收集结果。 - 其次,该问题可以回头选择元素。比如:当前track中的元素为[2],那么它可以回头选择1,使track变为[2, 1]。所以
start
变量在这里就不适用了。但是我们需要判断那个元素是使用过的,如果使用过那么就不可以再使用该元素。在该问题中,使用的是boolean[] used
数组来记录相应位置是否用过。
对于有重复元素的问题:
子集问题
如:nums = [1,2,2]
应该返回[ [],[1],[2],[1,2],[2,2],[1,2,2] ]
对于该问题,数组中有重复元素,元素不可以复选。
- 首先对于该问题,不可以回头选择元素,所以要定义
start
变量进行去重操作。 - 其次,还有可能重复的可能。比如:
track
中元素为[1]
时,我们对于数组中第一个2和第二个2分别进行选择就会出现两个[1, 2]
。出现重复的情况,所以需要进行去重。去重逻辑为
if(i > start && nums[i] == nums[i - 1])
这个主要思考为什么是nums[i] == nums[i - 1]
而不是nums[i] == nums[i + 1]
。因为用后者的话明显会出现索引越界的情况。即当i = nums.length - 1
时。
排列问题
如: nums = [1,2,2]
应该返回[ [1,2,2],[2,1,2],[2,2,1] ]
这次直接上代码:
List<List<Integer>> res = new LinkedList<>();
LinkedList<Integer> track = new LinkedList<>();
boolean[] used;
public List<List<Integer>> permuteUnique(int[] nums) {
// 先排序,让相同的元素靠在一起
Arrays.sort(nums);
used = new boolean[nums.length];
backtrack(nums);
return res;
}
void backtrack(int[] nums) {
if (track.size() == nums.length) {
res.add(new LinkedList(track));
return;
}
for (int i = 0; i < nums.length; i++) {
if (used[i]) {
continue;
}
// 新添加的剪枝逻辑,固定相同的元素在排列中的相对位置
if (i > 0 && nums[i] == nums[i - 1] && !used[i - 1]) {
continue;
}
track.add(nums[i]);
used[i] = true;
backtrack(nums);
track.removeLast();
used[i] = false;
}
}
着重分析 if (i > 0 && nums[i] == nums[i - 1] && !used[i - 1]) { continue; }
其他都较简单。
对于 nums = [1,2,2]
,我们可能会按照不同的顺序选择这两个2,所以会出现重复的情况:[1, 2, 2] [1, 2, 2]
。为了杜绝这样情况,我们就需要给相同的元素规定一个顺序,只有特定的顺序情况下,才是合法的,这样就不会出现重复的情况。对于 nums = [1,2,2]
我们可以规定,只有在第一个2加入了track
中后,第二个2才可以加入,这样就避免了重复。所以对于
if (i > 0 && nums[i] == nums[i - 1] && !used[i - 1]) { continue; }
也就比较好理解了,当元素相同,而前一个没有加入track
中时(!used[i - 1]
),该结果不可行,跳过。
总结
- 无论对于子集还是排列问题。对于去重时,首先判断是否可以回头选择元素,如果不可以回头选择,那么就需要
start
变量。如果可以回头选择元素,那么就需要用boolean[] used
数组来进行去重操作。 - 对于有重复元素的情况,需要进行不同的去重操作。