代码随想录算法训练营第24天 | 复习组合问题

2024年7月26日

题78. 子集
对于每个元素，都有选或者不选两种选择

class Solution {

    List<List<Integer>> res;
    List<Integer> path;
    int[] nums;

    public List<List<Integer>> subsets(int[] nums) {
        path = new ArrayList<>();
        res = new ArrayList<>();
        this.nums = nums;
        //每个元素，都有选或者不选
        backTracking(0);
        return res;
    }

    //index指当前到哪个元素了
    public void backTracking(int index){
        if(index==nums.length){
            res.add(new ArrayList<>(path));
        }else{
            path.add(nums[index]);
            backTracking(index+1);
            path.removeLast();
            backTracking(index+1);
        }
    }
}

题90. 子集 II
注意如果直接用list的包含方法来判断去重，必须要先对数组进行排序。

class Solution {

    List<List<Integer>> res;
    List<Integer> path;
    int[] nums;

    public List<List<Integer>> subsetsWithDup(int[] nums) {
        Arrays.sort(nums);
        path = new ArrayList<>();
        res = new ArrayList<>();
        this.nums = nums;
        //每个元素，都有选或者不选
        backTracking(0);
        return res;
    }

    //index指当前到哪个元素了
    public void backTracking(int index){
        if(index==nums.length){
            if(!res.contains(new ArrayList<>(path))){
                res.add(new ArrayList<>(path));
            }
        }else{
            backTracking(index+1);
            path.add(nums[index]);
            backTracking(index+1);
            path.removeLast();
        }
    }
}