首页 > 其他分享 >回溯 子集问题

回溯 子集问题

时间:2022-10-14 10:59:47浏览次数:76  
标签:used nums 问题 startIndex vector 子集 回溯 path

子集问题的特征相当于求遍所有的叶子节点,所以与前边的组合问题,分割问题,都可以抽象为收集树的节点的问题,
区别是子集是求所有的叶子节点;

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,还有一个就是记录初始位置的startIndex,只有当startIndex大于数组
的长度的时候,遍历才会终止,转换为代码如下:

if(startIndex >= nums.size())
{
    return;
}

递归函数与终止条件

终止条件如上所述。

void backtracking(vector<int>& nums,int startIndex)

单层遍历与回溯

因为可以不考虑终止条件,所以逻辑上相当于直接遍历整棵树

 for(int i = startIndex;i < nums.size();i++)
       {
           path.push_back(nums[i]); //子集收集元素
           backtracking(nums,i+1); //注意从i+1开始,不重复取元素
           path.pop_back();
       }

完整代码

class Solution {
private:
   vector<vector<int>> result;//存储所有的结果
   vector<int> path;
   void backtracking(vector<int>& nums,int startIndex)
   {
       result.push_back(path);//收集子集,要放在终止条件的上面,否则会遗漏掉元素
       if(startIndex >= nums.size())
       {
           return;
       }
       for(int i = startIndex;i < nums.size();i++)
       {
           path.push_back(nums[i]); //子集收集元素
           backtracking(nums,i+1); //注意从i+1开始,不重复取元素
           path.pop_back();
       }
   }
public:
    vector<vector<int>> subsets(vector<int>& nums) {
      result.clear();
      path.clear();
      backtracking(nums,0);
      return result;
    }
};

leetcode 90

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

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

示例 1:

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

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

思路

本题就在上边的题目的基础上,多了个不包含重复元素,其实就多了一个
去重的问题。

for (int i= startIndex;i < nums.size();i++)
        {
            if(i > 0 && nums[i] == nums[i-1] && used[i-1] == false)
            {//如果used[i-1]==true,说明同一树枝使用过candidates[i-1]
             //如果used[i-1]==false,说明同一树枝使用过candidates[i-1]
             //而我们要跳过同一树层使用过的元素
                continue;
            }
            path.push_back(nums[i]);
            used[i] = true;
            backtracking(nums,i+1,used);
            used[i] = false;
            path.pop_back();

完整代码

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(vector<int>& nums,int startIndex,vector<bool>& used)
    {
        result.push_back(path);
        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]);
            used[i] = true;
            backtracking(nums,i+1,used);
            used[i] = false;
            path.pop_back();
        }
    }
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
      result.clear();
      path.clear();
      vector<bool> used(nums.size(),false);
      sort(nums.begin(),nums.end()); //去重排序
      backtracking(nums,0,used);
      return result;
    }
};

标签:used,nums,问题,startIndex,vector,子集,回溯,path
From: https://www.cnblogs.com/7xiaomao/p/16790859.html

相关文章