LeetCode Subsets && Subsets II

Leetcode subset 问题

Subset tricky 解法

无重复： 子集数量一定是 (1 << nums.size())，所以可以根据每个集合的01编码解析是否包含这个元素。

如5 = 0101，则nums[0]和nums[2]在第5个集合中。类似的可以写出如下代码

class Solution {
public:
# define VVI vector<vector<int>>
# define VI vector<int>

    VVI subsets(VI& nums) {
        int len = nums.size();
        int total = 1 << len;
        VVI re(total, VI());
        for (int i = 0; i < len; ++i)
            for (int j = 0; j < total; ++j) 
                if ((j >> i) & 1) re[j].push_back(nums[i]);
        return re;
    }
};

Subset && Subset II search 解法

search时存在重复则跳过。关键在于代码行

if (i != pos && nums[i - 1] == nums[i]) continue;

可以解决重复问题。如果是当前循环中第一次遇到，则可以加入，保证cur集合中可以存在一定数量的重复元素。Subset和Subset II都可以通过这个解法解出来。

class Solution {
# define VVI vector<vector<int>>
# define VI vector<int>
# define pb push_back
# define po pop_back

# define LN ListNode

public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        VVI re;
        VI cur;

        search(nums, 0, cur, re);
        return re;
    }

    void search(VI& nums, int pos, VI cur, VVI& re) {
        if (pos == nums.size()) {
            re.pb(cur);
            return;
        }

        for (int i = pos; i < nums.size(); ++i) {
            if (i != pos && nums[i - 1] == nums[i]) continue; // for `Subset II`, skip duplicate
            cur.pb(nums[i]);
            search(nums, i + 1, cur, re);
            cur.po();
        }
        re.pb(cur);
    }
};