程式碼隨想錄演算法訓練營第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();
        }
    }
}