力扣 78. 子集 & 90. 子集 II

78题目描述:

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

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

解析:

回溯法

Java:

class Solution {
    public List<List<Integer>> subsets(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        List<Integer> list = new ArrayList<>();
        backTrack(res, list, nums, 0);
        return res;
    }
    public void backTrack(List<List<Integer>> res, List<Integer> list, int[] nums, int index) {
        res.add(new ArrayList<>(list));
        for(int i = index; i < nums.length; i++) {
            list.add(nums[i]);
            backTrack(res, list, nums, i + 1);
            list.remove(list.size() - 1);
        }
    }
}

JavaScript:

let res = [];
let list = [];
var subsets = function(nums) {
    res = [];
    backTrack(nums, 0);
    return res;

    function backTrack(nums, index) {
        res.push([...list]);
        for(let i = index; i < nums.length; i++) {
            list.push(nums[i]);
            backTrack(nums, i + 1);
            list.pop();
        }
    }
};

90题目描述:

给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。

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

解析:

回溯法

Java:

class Solution {
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        List<Integer> path = new ArrayList<>();
        Arrays.sort(nums);
        backTrack(res, path, nums, 0);
        return res;
    }
    public void backTrack(List<List<Integer>> res, List<Integer> path, int[] nums, int index) {
        res.add(new ArrayList<>(path));
        if(index == nums.length) {
            return;
        }
        for(int i = index; i < nums.length; i++) {
            if(i > index && nums[i] == nums[i - 1]) {
                continue;
            }
            path.add(nums[i]);
            backTrack(res, path, nums, i + 1);
            path.remove(path.size() - 1);
        }
    }
}

JavaScript:

var subsetsWithDup = function(nums) {
    let res = [], path = [];
    nums.sort((a,b) => a-b);
    backTrack(nums, 0);
    return res;

    function backTrack(nums, index) {
        res.push([...path]);
        if(index === nums.length) {
            return;
        }
        for(let i = index; i < nums.length; i++) {
            if(i > index && nums[i] === nums[i - 1]) {
                continue;
            }
            path.push(nums[i]);
            backTrack(nums, i + 1);
            path.pop();
        }
    }
};
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务