给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
示例 1:
输入:nums = [1,2,3]
输出:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]
思路:本题使用回溯算法,for循环横向遍历,递归纵向遍历,把它想象成树状结构,收集每一个节点值。定义全局变量result为结果集,path为收集到的子集。
- 确定回溯函数参数和返回值:回溯函数的参数有集合nums,每一层for循环起始搜索位置startIndex。回溯函数的返回类型一般都会void类型。
- 确定回溯函数的终止条件:如果startIndex大于等于nums.size(),说明搜索到了叶节点,return。
- 确定单层搜索的过程:for循环横向遍历,递归过程纵向遍历,for循环每次从startIndex开始遍历,然后用path保存每次取到的节点,result将path加入结果集,递归函数不断调用自己往深处遍历,到达叶子节点,就要返回。然后递归函数下面要写回溯部分,撤掉本次处理的结果,path弹出最后一个值。
因为把result.push_back(path);写到了path.push_back(nums[i]);的后面,所以空集是没有被加入到结果集的,要先将空集加进去。
代码:
class Solution { //78. 子集
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex) {
if (startIndex >= nums.size()) return;
for (int i = startIndex; i < nums.size(); ++i) {
path.push_back(nums[i]);
result.push_back(path);
backtracking(nums, i + 1);
path.pop_back();
}
}
vector<vector<int>> subsets(vector<int>& nums) {
result.push_back(path);
backtracking(nums, 0);
return result;
}
};