LeetCode Subsets && Subsets II

Leetcode subset 问题

Leetcode subset 问题
Subset tricky 解法

Subset Subset II search 解法

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