78. Subsets
这道题是一个很简单的利用回溯法求解的问题,如果想怎么增加题目的难度,那么如果所给集合中有重复的数字,那么相对于来说稍微复杂点,我们之后在讨论,对于程序员来说直接上代码比我废话要好很多。
class Solution {
public:
vector<vector<int>> subsets(vector<int>& nums) {
int m = nums.size();
vector<vector<int>> res;
vector<int> temp;
if (m == 0)
return res;
subsets(nums, res, temp, 0);
return res;
}
private:
void subsets(vector<int>& nums, vector<vector<int>>& res, vector<int>& temp, int start)
{
res.push_back(temp);
if (start == nums.size())
return;
for (int i = start; i < nums.size(); i++)
{
temp.push_back(nums[i]);
subsets(nums, res, temp, i + 1);
temp.pop_back();
}
return;
}
};
那么终止条件则是start == nums.size()。之后for循环依次遍历,非常好理解吧。那我们来看看一个进阶版本,稍微需要怎么增加一点判断。
90. Subsets II
废话不多说直接上代码:
class Solution {
public:
vector<vector<int>> subsets(vector<int>& nums) {
int m = nums.size();
vector<vector<int>> res;
vector<int> temp;
if (m == 0)
return res;
sort(nums.begin(), nums.end());
SameValue(nums, res, temp, 0);
return res;
}
private:
void SameValue(vector<int>& nums, vector<vector<int>>& res, vector<int>& temp, int start)
{
res.push_back(temp);
if (start == nums.size())
return;
for (int i = start; i < nums.size(); i++)
{
if (i == start || nums[i] != nums[i - 1])
{
temp.push_back(nums[i]);
SameValue(nums, res, temp, i + 1);
temp.pop_back();
}
}
return;
}
};
首先需要对集合进行排序,之后的区别就是判断当前未位置是否在start或者当前元素是否等于前一个。