【LeetCode】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 中的所有元素 互不相同

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subsets

法I：回溯法

还是比较简单的，直接上代码吧：

代码：

//回溯法：每一层代表选或者不选
class Solution {
public:
	vector<vector<int>> subsets(vector<int>& nums) {
		vector<vector<int>> re;
		vector<int> singleRe;
		int dimension = 0;
		traceback(dimension, nums, re, singleRe);
		return re;
	}

	void traceback(int dimension, vector<int> &nums, vector<vector<int>> &re, vector<int> &singleRe) {
		if (dimension == nums.size()) {
			re.push_back(singleRe);
			return;
		}
		for (int i = 0; i <= 1; i++) {//遍历该节点的子集：选或者不选该元素
			if (i == 0) {//不选
				traceback(dimension + 1, nums, re, singleRe);
			}
			else {//选
				singleRe.push_back(nums[dimension]);
				traceback(dimension + 1, nums, re, singleRe);
				singleRe.pop_back();
			}
		}
		return;
	}
};

执行结果：

法II：迭代法

上一下答案中的迭代求解的代码：

class Solution {
public:
    vector<int> t;
    vector<vector<int>> ans;

    vector<vector<int>> subsets(vector<int>& nums) {
        int n = nums.size();
        for (int mask = 0; mask < (1 << n); ++mask) {
            t.clear();
            for (int i = 0; i < n; ++i) {
                if (mask & (1 << i)) {
                    t.push_back(nums[i]);
                }
            }
            ans.push_back(t);
        }
        return ans;
    }
};