leetcode-子集

392 阅读1分钟

这是我参与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. 如果第一个数选 1,选第二个数,2、3 可选;
  2. 如果第一个数选 2,选第二个数,只有 3 可选(不能选1,产生重复组合)
  3. 如果第一个数选 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;
};