回溯算法之-子集

关于回溯法的模版请看:https://www.jianshu.com/p/2a9856b96a86

leetcode 78 子集

给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。解集 不能 包含重复的子集。你可以按 任意顺序 返回解集
这个题和组合总和的题类似,只不过我们将解集收集的地方不同,套回溯法模版

public List> subsets(int[] nums) {
        List> res = new ArrayList<>();
        Arrays.sort(nums);
        subsetsHelp(res, new ArrayList<>(), nums, 0);
        return res;
    }

    private void subsetsHelp(List> res, ArrayList list, int[] nums, int index) {
      // 在递归开始的地方将路径进行收集,一边回溯一边收集
        res.add(new ArrayList<>(list));
        for (int i = index; i < nums.length; i++) {
            list.add(nums[I]);
            subsetsHelp(res, list, nums, i + 1);
            list.remove(list.size()-1);
        }   
    }

Leetcode 90 子集2

给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
该题的解法和组合总和2基本相同

public List> subsetsWithDup(int[] nums) {
        List> res = new ArrayList<>();
        Arrays.sort(nums);
        subsetsWithDupHelp(res, new ArrayList<>(), nums, 0);
        return res;
    }

    private void subsetsWithDupHelp(List> res, ArrayList list, int[] nums, int index) {
        res.add(new ArrayList<>(list));
        for (int i = index; i < nums.length; i++) {
            // 同一层若相同元素被使用过则跳过
            if (i > index && nums[i] == nums[i-1]) {
                continue;
            }
            list.add(nums[I]);
            subsetsWithDupHelp(res, list, nums, i+1);
            list.remove(list.size()-1);
        }
    }

你可能感兴趣的:(回溯算法之-子集)