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);
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值