78.子集-中等
关键词:回溯算法、集合
题目描述:
给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: nums = [1,2,3]
输出:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
题解:
基本就是官方的答案了,不过还是记录一下
- 方法一:迭代法
即通过遍历的方式添加子集到ans集合中。
不过遍历的方式为,000,001,…,111。故而在函数里面需要不停地移位。
还有一个需要注意的地方:就是在res.add里面需要把子集t重新new一遍。因为java里存储的是t的地址,而不是将t的数据加到res里。
代码如下:
class Solution {
List<Integer> t = new ArrayList<Integer>();
List<List<Integer>> ans = new ArrayList<List<Integer>>();
public List<List<Integer>> subsets(int[] nums) {
int n = nums.length;
for (int mask = 0; mask < (1 << n); ++mask) {
t.clear();
for (int i = 0; i < n; ++i) {
if ((mask & (1 << i)) != 0) {
t.add(nums[i]);
}
}
ans.add(new ArrayList<Integer>(t));
}
return ans;
}
}
- 方法2:
回溯递归法
引用LeetCode评论解析一句话:
其实回溯算法关键在于:不合适就退回上一步,然后通过约束条件, 减少时间复杂度
class Solution {
public List<List<Integer>> subsets(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
backtrack(0, nums, res, new ArrayList<Integer>());
return res;
}
private void backtrack(int i, int[] nums, List<List<Integer>> res, ArrayList<Integer> tmp) {
res.add(new ArrayList<>(tmp));
for (int j = i; j < nums.length; j++) {
tmp.add(nums[j]);
backtrack(j + 1, nums, res, tmp);
tmp.remove(tmp.size() - 1);
}
}
}
参考
[1]:https://leetcode-cn.com/problems/subsets/solution/zi-ji-by-leetcode-solution/
[2]:https://leetcode-cn.com/problems/subsets/solution/hui-su-suan-fa-by-powcai-5/