代码题(18)— 子集

1、78. 子集

给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。

说明:解集不能包含重复的子集。

示例:

输入: nums = [1,2,3]
输出:
[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

方法1:使用循环

  我们可以一位一位的往上叠加,比如对于题目中给的例子[1,2,3]来说,最开始是空集,那么我们现在要处理1,就在空集上加1,为[1],现在我们有两个自己[]和[1],下面我们来处理2,我们在之前的子集基础上,每个都加个2,可以分别得到[2],[1, 2],那么现在所有的子集合为[], [1], [2], [1, 2],同理处理3的情况可得[3], [1, 3], [2, 3], [1, 2, 3], 再加上之前的子集就是所有的子集合了,代码如下:

以集合{1,2,3}为例。

初始化:[[]]

第一步:[[],[1]](在上面的结果的子集上添加元素1并上面的集合)

第二步:[[],[1],[2],[1,2]](上第二步结果上每个子集添加元素2并上面的集合)

第三步:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]](…)

整个添加的顺序为:

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

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>> res;
        vector<int> temp;
        sort(nums.begin(), nums.end());
        res.push_back(temp);
        for (int i = 0; i < nums.size(); ++i) {
            int size = res.size();
            for (int j = 0; j < size; ++j) {
                res.push_back(res[j]);//加入一个之前有的
                res.back().push_back(nums[i]);//对最后加入的那个数组插入一个数
            }
        }
        return res;
    }
};

方法2:

  下面来看递归的解法,相当于一种深度优先搜索。由于原集合每一个数字只有两种状态,要么存在,要么不存在,那么在构造子集时就有选择和不选择两种情况,所以可以构造一棵二叉树,左子树表示选择该层处理的节点,右子树表示不选择,最终的叶节点就是所有子集合,树的结构如下:

整个添加的顺序为:

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

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>> res;
        vector<int> temp;
        sort(nums.begin(), nums.end());
        getSub(nums, 0, temp, res);
        return res;
        
    }
    void getSub(vector<int> &nums, int pos, vector<int> &temp, vector<vector<int>> &res)
    {
        res.push_back(temp);
        for(int i=pos;i<nums.size();++i)
        {
            temp.push_back(nums[i]);
            getSub(nums, i + 1, temp, res);
            temp.pop_back();
        }
    }
};

方法三: 

  把数组中所有的数分配一个状态,true表示这个数在子集中出现,false表示在子集中不出现,那么对于一个长度为n的数组,每个数字都有出现与不出现两种情况,所以共有2n中情况,那么我们把每种情况都转换出来就是子集了,我们还是用题目中的例子, [1 2 3]这个数组共有8个子集,每个子集的序号的二进制表示,把是1的位对应原数组中的数字取出来就是一个子集,八种情况都取出来就是所有的子集了。

2、90. 子集 II

给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。

说明:解集不能包含重复的子集。

示例:

输入: [1,2,2]
输出:
[
  [2],
  [1],
  [1,2,2],
  [2,2],
  [1,2],
  []
]

在处理到第二个2时,由于前面已经处理了一次2,这次我们只在添加过2的[2] 和 [1 2]后面添加2,其他的都不添加,那么这样构成的二叉树如下图所示:

复制代码
                        []        
                   /          \        
                  /            \     
                 /              \
              [1]                []
           /       \           /    \
          /         \         /      \        
       [1 2]       [1]       [2]     []
      /     \     /   \     /   \    / \
  [1 2 2] [1 2]  X   [1]  [2 2] [2] X  []
复制代码

代码只需在原有的基础上增加一句话,while (S[i] == S[i + 1]) ++i; 这句话的作用是跳过树中为X的叶节点,因为它们是重复的子集,应被抛弃。代码如下:

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<vector<int>> res;
        if(nums.empty())
            return res;
        vector<int> temp;
        sort(nums.begin(),nums.end());
        subsetsDfs(nums,0,temp,res);
        return res;
    }
    void subsetsDfs(vector<int>& nums, int pos, vector<int>& temp, vector<vector<int>> &res )
    {
        res.push_back(temp);
        for(int i=pos;i<nums.size();++i)
        {
            if(i>pos && nums[i] == nums[i-1])
                continue;
                
            temp.push_back(nums[i]);
            subsetsDfs(nums,i+1,temp,res);
            temp.pop_back();
        }
        
    }
};

 

posted @ 2018-07-10 11:48  深度机器学习  阅读(270)  评论(0编辑  收藏  举报