78.Subsets - medium
给定一个数组,求这个数组的子集,空集也算子集。
Input: nums = [1,2,3]
Output:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
思路:
其实和 77题 很类似,基本没什么变化,只需要在77题基础上稍作改动即可。比如每一次的 K 值。设一个临时容器 tmp ,以及一个记录起点的变量 start ,每一次从 start 开始循环,递归往里面放一个数,当容器 tmp 里面的个数等于 k 个时,就将这个容器写入到最终的结果容器中,然后返回上一层,再将这一层中放入的数字 pop_back( ),再重新放入下一个数字,不断的循环递归。
class Solution { public: vector<vector<int>> subsets(vector<int>& nums) { vector<vector<int>> res; int n = nums.size(); for (int k = 1; k <= n; k++) { vector<int> tmp; subsetsDFS(nums, 0, k, tmp, res); } res.push_back({}); return res; } void subsetsDFS(vector<int>& nums, int start, int k, vector<int>& tmp, vector<vector<int>>& res) { if (tmp.size() == k) { res.push_back(tmp); return; } for (int i = start; i < nums.size(); i++) { tmp.push_back(nums[i]); subsetsDFS(nums, i + 1, k, tmp, res); tmp.pop_back(); } } };
Java 版:
• 参考 “77.组合”,深度优先搜索 DFS;
• 在主函数中,循环要组合的长度为 0 - n 即可,其余完全和 77题一模一样。
class Solution { public List<List<Integer>> subsets(int[] nums) { List<List<Integer>> res = new ArrayList<>(); for(int i = 0; i <= nums.length; i++){ //枚举组合的长度 i DFS(new ArrayList<>(), nums, i, 0, res); //用这个长度去寻找答案 } return res; } public void DFS(List<Integer> tmp, int[] nums, int n, int start, List<List<Integer>> res){ if(tmp.size() == n){ res.add(new ArrayList(tmp)); //加入结果集 return; } for(int i = start; i < nums.length; i++){ tmp.add(nums[i]); DFS(tmp, nums, n, i+1, res); if(tmp.size() > 0) tmp.remove(tmp.size() - 1); } } }