回溯算法之-子集 leetcode 78 子集 Leetcode 90 子集2

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

leetcode 78 子集

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

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

    private void subsetsHelp(List<List<Integer>> res, ArrayList<Integer> 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<List<Integer>> subsetsWithDup(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        Arrays.sort(nums);
        subsetsWithDupHelp(res, new ArrayList<>(), nums, 0);
        return res;
    }

    private void subsetsWithDupHelp(List<List<Integer>> res, ArrayList<Integer> 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);
        }
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章