关闭

78. Subsets

71人阅读 评论(0) 收藏 举报
分类:

Given a set of distinct integers, nums, return all possible subsets.

Note: The solution set must not contain duplicate subsets.

For example,
If nums = [1,2,3], a solution is:

[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

分析:求集合的所有子集问题。题目要求子集中元素非递减序排列,因此我们先要对原来的集合进行排序。原集合中每一个元素在子集中有两种状态:要么存在、要么不存在。这样构造子集的过程中每个元素就有两种选择方法:选择、不选择,因此可以构造一颗二叉树,例如对于例子中给的集合[1,2,3],构造的二叉树如下(左子树表示选择该层处理的元素,右子树不选择),最后得到的叶子节点就是子集:
这里写图片描述

class Solution {
private:
    vector<vector<int>> res;
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        res.clear();
        sort(nums.begin(),nums.end());
        vector<int> tmp;
        dfs(nums,0,tmp);
        return res;
    }
private:
    void dfs(vector<int> nums,int index,vector<int>& tmp){
        if(index == nums.size()){
            res.push_back(tmp);
            return;
        }
        tmp.push_back(nums[index]);
        dfs(nums,index+1,tmp);
        tmp.pop_back();
        dfs(nums,index+1,tmp);
    }
};

类似的解法:以{1,2,3}为例,首先取出1,加入子集合,后面的2和3就是1的子节点,先取出2,把[1, 2]加入子集合,后面的3就是2的子节点,取出3,把[1, 2, 3]加入子集合。然后回溯,取出3,将[1, 3]加入子集合。
1处理完成之后,我们可以同样方式处理2,以及3。

class Solution {
public:
    vector<vector<int> > res;
    vector<vector<int> > subsets(vector<int> &S) {
        res.clear();
        vector<int> tmp;
        //sort(S.begin(),S.end());
        res.push_back(vector<int>());
        dfs(S,0,tmp);
        return res;
    }
    void dfs(vector<int> nums, int start, vector<int>& tmp){
        if(start == nums.size()){
            return ;
        }
        for(int i = start; i<nums.size(); i++){
            tmp.push_back(nums[i]);
            res.push_back(tmp);
            dfs(nums,i+1,tmp);
            tmp.pop_back();
        }
    }
};

subsets Ⅱ 是在subsetⅠ的基础上去重。思想类似。

class Solution {
public:
    vector<vector<int>> res;
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        res.clear();
        sort(nums.begin(),nums.end());
        res.push_back(vector<int>());
        vector<int> tmp;
        dfs(nums,0,tmp);
        return res;
    }
    void dfs(vector<int> nums, int start, vector<int> &tmp){
        if(start == nums.size()){
            return ;
        }
        vector<vector<int>>::iterator it = res.begin();
        for(int i = start; i<nums.size(); i++){
            tmp.push_back(nums[i]);
            res.push_back(tmp);
            dfs(nums,i+1,tmp);
            tmp.pop_back();
            while(i<nums.size()-1 && nums[i]==nums[i+1]){
                i++;
            }
        }
    }
};
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:2805次
    • 积分:150
    • 等级:
    • 排名:千里之外
    • 原创:10篇
    • 转载:4篇
    • 译文:1篇
    • 评论:0条
    文章分类