Leetcode 78.子集


题目描述:

给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。

解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

示例 1:

1
2
输入:nums = [1,2,3]
输出:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]

示例 2:

1
2
输入:nums = [0]
输出:[[],[0]]

提示:

  • 1 <= nums.length <= 10
  • -10 <= nums[i] <= 10
  • nums 中的所有元素 互不相同

链接:

https://leetcode-cn.com/problems/subsets


题目分析

  对于 nums 中的每个数,选择或者不选择都是两种不同的情况,也即幂集的个数为 $2^n$。我们可以使用回溯的方法,不选择 nums[i] 进行递归,得到一种结果后回溯,选择 nums[i] 进行递归。每次遍历完 nums[i] 即得到一个结果,将其加入到结果数组中。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
class Solution {
void backtrack(vector<int>& nums, vector<vector<int>>& result, vector<int>& now, int index){
if(index == nums.size()){
result.push_back(now);
return;
}
backtrack(nums, result, now, index+1);
now.push_back(nums[index]);
backtrack(nums, result, now, index+1);
now.pop_back();
}
public:
vector<vector<int>> subsets(vector<int>& nums) {
vector<vector<int>> result;
vector<int> now;
backtrack(nums, result, now, 0);
return result;
}
};

  时间复杂度:$O(n\times2^n)$,其中 $n$ 为 nums 数组的大小。因为总共有 $2^n$ 种结果,而对于每种结果我们需要 $O(n)$ 的时间将其加入到答案中。
  空间复杂度:$O(n)$,其中 $n$ 为 nums 数组的大小。递归的最大深度为 $n$。