leetcode每日一题35

90. 子集 II

回溯嘛

子集啊排列组合啊棋盘啊都是回溯

回溯三部曲走起

跟78.子集比,本题给出的数组里存在重复元素了

所以在取元素时,如果同一层里取过某个元素,那么在该层就不能取重复的该元素了

如给出的数组[1,2,2]

可以在某一次递归中第一个取2放进子集,但后面的递归就不允许第一个取2放进子集里了

详情可以看代码随想录的图
代码随想录

所以要有一个数组used记录该层里取过的数

  1. 递归函数参数
    回溯问题一般涉及两个全局变量:
    保存本次递归中符合条件的结果path
    保存所有符合条件的结果的集合result
    以及回溯函数backtracking,因为是求子集问题,所以取过的元素不能重复取,所以回溯时,for循环要从startIndex开始,而不是从0开始
cpp 复制代码
vector<int> path;
vector<vector<int>> result;
void backtracking(vector<int>& nums, int startIndex, vector<bool>& used)
  1. 递归终止条件
    当此时的startIndex已经大于数组长度时,就没有没取过的数组元素了,本次递归就终止了
cpp 复制代码
if(startIndex>=nums.size()){
	return;
}
  1. 单层搜索逻辑
    单层的搜索逻辑是
    先将取出来的数存入path,再递归调用自身,然后回溯,删掉刚才取出来的数
cpp 复制代码
path.push_back(nums[i]);
backtracking(......);
path.pop_back();

本题中,要判断取的nums[i]有没有使用过

如果没有,那么在backtracking要传入used数组,所以要递归前标记nums[i]已经被使用过了而递归后,需要回溯,从path中删除nums[i],所以要恢复为nums[i]未被使用

cpp 复制代码
if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
                continue;
}//判定nums[i]有没有使用过
path.push_back(nums[i]);
used[i]=true;
backtracking(nums, i+1,used);
used[i]=false;
path.pop_back();

所以,回溯算法模板为

cpp 复制代码
void backtracking(参数) {
		收集子集result.push_back(path);
    if (终止条件) {
        存放结果;
        return;
    }

    for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
        处理节点;
        backtracking(路径,选择列表); // 递归
        回溯,撤销处理结果
    }
}

那么组合起来,本题的回溯函数为

cpp 复制代码
vector<int> path;
vector<vector<int>> result;
void backtracking(vector<int>& nums, int startIndex, vector<bool>& used){
	result.push_back(path);//收集子集
	if(startIndex>=nums.size()){
		return;
	}
	for(int i =startIndex;i<nums.size();i++)
	{
		if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
	                continue;
		}//判定nums[i]有没有使用过
		path.push_back(nums[i]);
		used[i]=true;
		backtracking(nums, i+1,used);
		used[i]=false;
		path.pop_back();
	}
}
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;
    }

整理一下,得到最终代码:

cpp 复制代码
class Solution {
private:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& nums, int startIndex, vector<bool>& used){
        result.push_back(path);//收集子集,要放在判定停止条件前,防止漏数
        if(startIndex>=nums.size()){
            return;
        }
        for(int i =startIndex;i<nums.size();i++)
        {
            if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
                        continue;
            }//判定nums[i]有没有使用过
            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;
    }
};
相关推荐
进击的荆棘几秒前
C++起始之路——类和对象(下)
开发语言·c++
liu****6 分钟前
10.排序
c语言·开发语言·数据结构·c++·算法·排序算法
快乐的划水a13 分钟前
std::thread与pthread关系
c++
_OP_CHEN14 分钟前
【算法基础篇】(三十二)动态规划之背包问题扩展:从多重到多维,解锁背包问题全场景
c++·算法·蓝桥杯·动态规划·背包问题·算法竞赛·acm/icpc
listhi52017 分钟前
机械系统运动学与动力学在MATLAB及SimMechanics中的实现方案
人工智能·算法·matlab
fufu031118 分钟前
Linux环境下的C语言编程(三十九)
c语言·数据结构·算法·链表
炽烈小老头19 分钟前
【 每天学习一点算法 2025/12/12】回文链表
学习·算法·链表
前端小L19 分钟前
回溯算法专题(十):二维递归的完全体——暴力破解「解数独」
数据结构·算法
AI科技星1 小时前
姬无烦科幻与张祥前统一场论的完美融合
数据结构·人工智能·算法·机器学习·重构
Studying 开龙wu1 小时前
Windos 10系统安装OpenPose的CPU版本过程说明和Release版本直接使用
c++·windows