这是我参与11月更文挑战的第7天,活动详情查看:2021最后一次更文挑战
题目描述
给你一个整数数组 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` 中的所有元素 **互不相同**
实现思路
解法一:使用回溯算法来解决,定义结果集res通过遍历nums通过回调函数fn来寻找以nums[index]开头的所有结果并且添加进res。
用for枚举出当前可选的数,比如选第一个数时:1、2、3 可选。
- 如果第一个数选 1,选第二个数,2、3 可选;
- 如果第一个数选 2,选第二个数,只有 3 可选(不能选1,产生重复组合)
- 如果第一个数选 3,没有第二个数可选
/**
* @param {number[]} nums
* @return {number[][]}
*/
var subsets = function(nums) {
let len = nums.length
let res = []
function fn(index, list) {
res.push(list)
for (let i = index; i < len; i++) {
fn(i+1, [...list, nums[i]])
}
}
fn(0, [])
return res
};
解法二:使用位运算 由题可知返回该数组所有可能的子集(幂集),所以我们可以直接得到返回数组的长度nums.length次方,遍历,根据数字位置来找子集,这里位运算在这里的思想主要是,数组中的每一个数字都有选和不选两种状态,用0和1表示,如果数字i的某一个位置是1,就把数组中对应的数字添加到集合。
/**
* @param {number[]} nums
* @return {number[][]}
*/
var subsets = function(nums) {
// 子集长度为2的nums.length次方,所以通过位运算移位
let len = 1 << nums.length;
let res = [];
// 遍历,根据数字位置来找子集
for(let i=0;i<len;i++){
let list = [];
for(let j=0;j<nums.length;j++){
if(((i>>j)&1) == 1){
list.push(nums[j])
}
}
res.push(list);
}
return res;
};