代码随想录第二十五天 78.子集 90.子集II 491.非递减子序列

时间:2024-03-02 17:37:31

LeetCode 78 子集

题目描述

给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。

解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

示例 1:

输入:nums = [1,2,3]
输出:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]

示例 2:

输入:nums = [0]
输出:[[],[0]]

思路

        这道题目和普通的回溯问题的区别在于,每加入一次新的元素都要将结果加入result数组中,而不是需要判断path数组中的元素是否达到题目要求的标准,才加入result数组,所以做出的改变就是,在for循环的过程中,每一次循环都需要加当前的数组加入result中。

代码实现

class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& nums,int startindex){
        if(path.size() == 0) result.push_back(path);
        if(path.size() == nums.size()){
            return;
        }
        for(int i = startindex;i < nums.size();i++){
            path.push_back(nums[i]);
            result.push_back(path);
            backtracking(nums,i + 1);
            path.pop_back();
        }
        return;
    }
    vector<vector<int>> subsets(vector<int>& nums) {
        backtracking(nums,0);
        return result;
    }
};

LeetCode 90 子集II 

题目描述

给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。

解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。

示例 1:

输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]

示例 2:

输入:nums = [0]
输出:[[],[0]]

思路

        子集一问题和子集二问题的区别在于子集二需要进行剪枝。而如何进行剪枝和前面的组合问题是一样的步骤。要去重的是“同一树层上的使用过”,如何判断同一树层上元素(相同的元素)是否使用过了呢。

如果nums[i] == nums[i - 1] 并且 used[i - 1] == false,就说明:前一个树枝,使用了nums[i - 1],也就是说同一树层使用过nums[i - 1]

此时for循环里就应该做continue的操作。

这块比较抽象,如图:

40.组合总和II1

可以看出在nums[i] == nums[i - 1]相同的情况下:

  • used[i - 1] == true,说明同一树枝nums[i - 1]使用过
  • used[i - 1] == false,说明同一树层nums[i - 1]使用过

used[i - 1] = false说明已经进入另一个树枝,所以前一个数才会被跳过,没有遍历到。而 used[i - 1] == true,说明是进入下一层递归,去下一个数,所以是树枝上,如图所示:

代码实现

class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& nums,int startindex,vector<bool>& used){
        if(path.size() == 0) result.push_back(path);
        if(path.size() == nums.size()) return;
        for(int i = startindex;i < nums.size();i++){
            if(i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false){
                continue;
            }
            path.push_back(nums[i]);
            result.push_back(path);
            used[i] = true;
            backtracking(nums,i + 1,used);
            path.pop_back();
            used[i] = false;
        }
        return;
    }
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<bool> used(nums.size(),false);
        sort(nums.begin(),nums.end());
        backtracking(nums,0,used);
        return result;
    }
};

 LeetCode 491 非递减子序列

题目描述

给你一个整数数组 nums ,找出并返回所有该数组中不同的递增子序列,递增子序列中 至少有两个元素 。你可以按 任意顺序 返回答案。

数组中可能含有重复元素,如出现两个整数相等,也可以视作递增序列的一种特殊情况。

示例 1:

输入:nums = [4,6,7,7]
输出:[[4,6],[4,6,7],[4,6,7,7],[4,7],[4,7,7],[6,7],[6,7,7],[7,7]]

示例 2:

输入:nums = [4,4,3,2,1]
输出:[[4,4]]

思路

  • 递归函数参数

本题求子序列,很明显一个元素不能重复使用,所以需要startIndex,调整下一层递归的起始位置。

代码如下:

vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex)
  • 终止条件

本题其实类似求子集问题,也是要遍历树形结构找每一个节点,所以可以不加终止条件,startIndex每次都会加1,并不会无限递归。

但本题收集结果有所不同,题目要求递增子序列大小至少为2,所以代码如下:

if (path.size() > 1) {
    result.push_back(path);
    // 注意这里不要加return,因为要取树上的所有节点
}
  • 单层搜索逻辑

491. 递增子序列1

 在图中可以看出,同一父节点下的同层上使用过的元素就不能再使用了。这里避免重复的思路是使用一个unoedered set去判断某一数字在当前树层是否使用过。

那么单层搜索代码如下:

unordered_set<int> uset; // 使用set来对本层元素进行去重
for (int i = startIndex; i < nums.size(); i++) {
    if ((!path.empty() && nums[i] < path.back())
            || uset.find(nums[i]) != uset.end()) {
            continue;
    }
    uset.insert(nums[i]); // 记录这个元素在本层用过了,本层后面不能再用了
    path.push_back(nums[i]);
    backtracking(nums, i + 1);
    path.pop_back();
}

代码实现

class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& nums,int startindex){
        if(path.size() > 1){
            result.push_back(path);
        }
        unordered_set<int> uset;
        for(int i = startindex;i < nums.size();i++){
            if((!path.empty() && nums[i] < path.back())
                || uset.find(nums[i]) != uset.end()){
                    continue;
            }
            uset.insert(nums[i]); 
            path.push_back(nums[i]);
            backtracking(nums, i + 1);
            path.pop_back();
        }
    }
    vector<vector<int>> findSubsequences(vector<int>& nums) {
        backtracking(nums, 0);
        return result;
    }
};