题目
法1:DFS
必须掌握的方法!!!
我们也可以用递归来实现子集枚举。
假设我们需要找到一个长度为 nnn 的序列 aaa 的所有子序列,代码框架是这样的:
void dfs(int cur, int n, LinkedList<Integer> tmp) {
if (cur == n) {
// 记录答案
// ...
return;
}
// 考虑选择当前位置
tmp.add(cur);
dfs(cur + 1, n, k);
tmp.removeLast();
// 考虑不选择当前位置
dfs(cur + 1, n, k);
}
答案如下:
class Solution {
public List<List<Integer>> subsets(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
if (nums.length == 0) {
return res;
}
LinkedList<Integer> tmp = new LinkedList<>();
dfs(nums, 0, tmp, res);
return res;
}
public void dfs(int[] nums, int startIndex, LinkedList<Integer> tmp, List<List<Integer>> res) {
if (startIndex == nums.length) {
res.add(new ArrayList<>(tmp));
return;
}
tmp.add(nums[startIndex]); // 选择当前值进行dfs
dfs(nums, startIndex + 1, tmp, res);
tmp.removeLast(); // 不选择当前值进行dfs
dfs(nums, startIndex + 1, tmp, res);
}
}
法2:动态规划
简单的dp问题。
class Solution {
public List<List<Integer>> subsets(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
if (nums.length == 0) {
return res;
}
res.add(new ArrayList<>());
for (int i = 0; i < nums.length; ++i) {
int size = res.size();
for (int k = 0; k < size; ++k) {
List<Integer> newList = new ArrayList<>(res.get(k));
newList.add(nums[i]);
res.add(newList);
}
}
return res;
}
}