题目描述:
给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
示例 1:
输入:nums = [1,2,3]
输出:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]
示例 2:
输入:nums = [0]
输出:[[],[0]]
提示:
1 <= nums.length <= 10
-10 <= nums[i] <= 10
nums 中的所有元素 互不相同
参考 labuladong 的算法小抄:的解法:
List<List<Integer>> res = new LinkedList<>();
// 记录回溯算法的递归路径
LinkedList<Integer> track = new LinkedList<>();
// 主函数
public List<List<Integer>> subsets(int[] nums) {
backtrack(nums, 0);
return res;
}
// 回溯算法核心函数,遍历子集问题的回溯树
void backtrack(int[] nums, int start) {
// 前序位置,每个节点的值都是一个子集
res.add(new LinkedList<>(track));
// 回溯算法标准框架
for (int i = start; i < nums.length; i++) {
// 做选择
track.addLast(nums[i]);
// 通过 start 参数控制树枝的遍历,避免产生重复的子集
backtrack(nums, i + 1);
// 撤销选择
track.removeLast();
}
}
相应的C++解法代码如下:
class Solution {
private:
// 所有冥集结果
vector<vector<int>> res;
// 当前子集(路径)
vector<int> track;
public:
vector<vector<int>> subsets(vector<int>& nums) {
backtrack(nums, 0);
return res;
}
void backtrack(vector<int>& nums, int start) {
res.push_back(track);
// 回溯算法核心框架
// 遍历选择列表
for (int i = start; i < nums.size(); i++) {
// 做选择
track.push_back(nums[i]);
backtrack(nums, i+1);
// 撤销选择
track.pop_back();
}
}
};
完整的测试示例C++代码如下:
/*
力扣78.子集
给你一个整数数组 nums ,数组中的元素 互不相同 。
返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
示例 1:
输入:nums = [1,2,3]
输出:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]
示例 2:
输入:nums = [0]
输出:[[],[0]]
提示:
1 <= nums.length <= 10
-10 <= nums[i] <= 10
nums 中的所有元素 互不相同
*/
#include <iostream>
#include <vector>
using namespace std;
class Solution {
private:
// 所有冥集结果
vector<vector<int>> res;
// 当前子集(路径)
vector<int> track;
public:
vector<vector<int>> subsets(vector<int>& nums) {
backtrack(nums, 0);
return res;
}
void backtrack(vector<int>& nums, int start) {
res.push_back(track);
// 回溯算法核心框架
// 遍历选择列表
for (int i = start; i < nums.size(); i++) {
// 做选择
track.push_back(nums[i]);
backtrack(nums, i + 1);
// 撤销选择
track.pop_back();
}
}
};
int main()
{
vector<int> nums = { 1, 2, 3 };
//vector<int> nums = { 0 };
Solution sol;
vector<vector<int>> res = sol.subsets(nums);
for (int i = 0; i < res.size(); i++) {
if (i == 0) {
std::cout << "[[";
} else {
std::cout << "[";
}
for (int j = 0; j < res[i].size(); j++) {
std::cout << res[i][j];
if (j < res[i].size() - 1) {
std::cout << ",";
}
}
if (i < res.size() - 1) {
std::cout << "],";
} else {
std::cout << "]]";
}
}
return 0;
}