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);
        }
    }
}

 

posted @ 2020-06-01 19:23  星海寻梦233  阅读(109)  评论(0编辑  收藏  举报