回溯算法之-子集 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);
        }
    }