90. 子集 II
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subsets-ii
前言
LeetCode也是刷了一个月了吧,每日一题。之前虽说有写题解,但总归还是有点水,可能更多的还是自己看的懂,所以说一直没发到博客上。现在的话多少也是对算法题有一定理解了吧,应该来说姑且还是能写一点东西出来了。
但实话说有时还是会有点懒,毕竟光写算法题就花了那么多时间了,要我再写篇博客、题解实话说还是有点难顶。而且我目前还是要多花时间去学Java,所以说可能就不太能认真写题解吧,大致把自己思路写出来可能就差不多了。
所以说我应该就是偶尔更新一下算法题吧,平时也就自己写一点思路,贴下代码就差不多了。如果说某些题目对我算法能力提升的比较多,那我可能就会认真写一篇题解了。不过话说回来这篇好像也没怎么写的明白= =。
题目
给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
示例 1:
输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]
给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
思路
这题也是用的回溯,回溯的话之前的一个题131. 分割回文串也是学到了很多,然后这题也是照着之前的思路去写的,按理说应该先做子集1,但我觉得我应该是能做出来的,最终虽说错了2次,但总归还是做出来了吧。
回溯就是循环+递归,先是深搜,深搜到最后加到 lists 里面,然后去掉 list 末尾的数,直接从下一个数继续深搜。核心逻辑就像这样:
public void buildList(int temp, int[] nums) {
lists.add(new ArrayList<>(list));
for (int i = temp; i < n; i++) {
list.add(nums[i]);
buildList(i + 1, nums);
list.remove(list.size() - 1);
}
}
多调试一下感受感受还是能想的出来的。
然后主要会碰到的问题就是会有重复的集合,要想办法去重。一种重复是像这样:[4,4,1] 和 [1,4,4] 虽然list会不同但实际在集合中是相同的;还有一种是像这样:[1, 4, 4, 4], [1, 4, 4, 4, 4], [1, 4, 4, 4],在递归的时候重复了。
第一种的话就是先对数组排下序,保证重复的元素都是相邻的,这样list就一定是相同的了;
第二种的话就是需要在递归时判断去重了,我本来是用的lists.contains(list)
方法去重的,然后时间15ms。。。。
然后看了1ms的代码也是发现直接先跳到重复项最后一个就行了= =。
然后题解也是有用二进制枚举的,思路是能理解吧,不过这种方法适用性似乎没有回溯强的样子,好像只能适用枚举的情况。不过姑且也是一种思路吧,吸收一下也是很好的。
代码
public class M3D31Q90 {
private List<List<Integer>> lists = new ArrayList<>();
private List<Integer> list = new ArrayList<>();
private int n;
public List<List<Integer>> subsetsWithDup(int[] nums) {
n = nums.length;
Arrays.sort(nums);
buildList(0, nums);
return lists;
}
public void buildList(int temp, int[] nums) {
lists.add(new ArrayList<>(list));
for (int i = temp; i < n; i++) {
if(i > temp && nums[i] == nums[i-1])
continue;
list.add(nums[i]);
buildList(i + 1, nums);
list.remove(list.size() - 1);
}
}
public static void main(String[] args) {
M3D31Q90 q = new M3D31Q90();
List<List<Integer>> lists;
lists = q.subsetsWithDup(new int[]{4, 4, 4, 1, 4});
System.out.println(lists);
}
}