代码随想录二刷 Day 30

一天24小时在图书馆学习2023-10-08 6:02

78.子集

与组合问题不同的就是要收集所有的子节点，而不仅仅的叶子节点；唯一一行代码的区别就是要把result的收集过程放到递归程序的最开始那里

cpp 复制代码

class Solution {
public:

    vector<vector<int>> result;
    vector<int> path;

    void traversal(vector<int>& nums, int start_index) {
        result.push_back(path); // 收集子集，要放在终止添加的上面，否则会漏掉自己; 这句是唯一和组合题目不一样的一句，放的位置不同就可以收集所有的节点
        if (start_index == nums.size()) return;

        for( int i = start_index; i <nums.size(); i++) {
            path.push_back(nums[i]);
            traversal(nums, i+1);
            path.pop_back();
        }
    }

    vector<vector<int>> subsets(vector<int>& nums) {
        traversal(nums, 0);
        return result;
    }
};

90.子集II

这个题和上一个题区别就是要加上树层的去重,整体还是挺好做的，有些细节还是不会写

cpp 复制代码

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;

    void traversal(vector<int>& nums, int start_index, vector<bool>& used) {
        result.push_back(path);
        if (start_index == nums.size()) { 
            return;
        }

        for( int i = start_index; i <nums.size(); i++) {
            //if( nums[i] == nums[i-1] && used[i-1] == false){这句不行i>0会报奇怪的错
            if(i>0 && nums[i] == nums[i-1] && used[i-1] == false){
                //return;//这句写错了，这里应该是跳出这个循环而不是return;
                continue;
            }
            path.push_back(nums[i]);
            used[i] = true;
            traversal(nums, i+1, used);
            path.pop_back();
            used[i] = false;
        }
    }

    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<bool> used(nums.size(), false); //这句自己还是写不出
        sort(nums.begin(),nums.end());
        traversal(nums, 0, used);
        return result;
    }
};