题目
给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
示例
输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]
输入:nums = [0]
输出:[[],[0]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subsets-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1:回溯
Java实现1
class Solution {
List<List<Integer>> res = new ArrayList<>();
// 使用 visited 数组来记录哪一个元素在当前路径中被使用了
boolean[] visited;
public List<List<Integer>> subsetsWithDup(int[] nums) {
//首先排序,让相同的两个元素排到一起去,便于去重
Arrays.sort(nums);
LinkedList<Integer> track = new LinkedList<>();
visited = new boolean[nums.length];
track_back(nums, 0, track);
return res;
}
public void track_back(int[] nums, int start, LinkedList<Integer> track){
res.add(new ArrayList<>(track));
for (int i = start; i < nums.length; i++){
//如果当前元素和前一个元素相同,而且前一个元素没有被访问,说明前一个相同的元素在当前层已经被用过了
//⭐注意:i > 0
if (i > 0 && nums[i - 1] == nums[i] && !visited[i - 1]) continue;
track.addLast(nums[i]);
visited[i] = true;
track_back(nums, i + 1, track);
track.removeLast();
visited[i] = false;
}
}
}
Java实现2:剪枝
class Solution {
List<List<Integer>> res = new ArrayList<>();
public List<List<Integer>> subsetsWithDup(int[] nums) {
LinkedList<Integer> track = new LinkedList<>();
Arrays.sort(nums);
track_back(track, nums, 0);
return res;
}
public void track_back(LinkedList<Integer> track, int[] nums, int start) {
res.add(new ArrayList<>(track));
for (int i = start; i < nums.length; i++) {
//⭐注意:i > start
if (i > start && nums[i - 1] == nums[i]) continue;
track.add(nums[i]);
track_back(track, nums, i + 1);
track.removeLast();
}
}
}
Java实现3
题目所说的重复是指:【4,4,1】 【4,1,4】就是重复
先将nums排序。再进行回溯。
class Solution {
List<List<Integer>> res = new ArrayList<>();
public List<List<Integer>> subsetsWithDup(int[] nums) {
Arrays.sort(nums); //⭐
LinkedList<Integer> track = new LinkedList<>();
track_back(nums, 0, track);
return res;
}
public void track_back(int[] nums, int start, LinkedList<Integer> track){
List<Integer> var = new ArrayList<>(track);
if (!res.contains(var)) res.add(var); //⭐
for (int i = start; i < nums.length; i++){
track.addLast(nums[i]);
track_back(nums, i + 1, track);
track.removeLast();
}
}
}