[Algorithm][回溯][组合][目标和][组合总和]详细讲解

目录


1.组合

1.题目链接


2.算法原理详解

  • 思路:每次都只选一个数,此后只能选它后面的数
  • 函数设计
    • 全局变量
      • vector<vector<int>> ret
      • vector<int> path
    • DFS()设计:void DFS(nums, pos)
    • 递归出口path.size() == k
    • 剪枝 :控制参数,每次从此位置下一个位置开始递归

3.代码实现

cpp 复制代码
class Solution 
{
    int _n;
    int _k;

    vector<vector<int>> ret;
    vector<int> path;
public:
    vector<vector<int>> combine(int n, int k) 
    {
        _n = n;
        _k = k;
        DFS(1);
        return ret;
    }

    void DFS(int start)
    {
        if(path.size() == _k)
        {
            ret.push_back(path);
        }

        // 递归 + 剪枝
        for(int i = start; i <= _n; i++)
        {
            path.push_back(i);
            DFS(i + 1);
            path.pop_back(); // 回溯,恢复现场
        }
    }
};

2.目标和

1.题目链接


2.算法原理详解

  • 本题与子集逻辑几乎相同
  • 本题会实现两种代码,可以通过这两种代码来感受:回溯的两种做法
    • path是全局变量的时候
      • 本题可能会超时
    • path作为参数
      • 此时编译器/代码会代为回溯,每次回溯都省去了一次加/减运算,故效率有所提高

3.代码实现

cpp 复制代码
// v1.0 效率低,可能会超时
class Solution 
{
    int ret = 0;
    int path = 0;
    int _target = 0;
public:
    int findTargetSumWays(vector<int>& nums, int target) 
    {
        _target = target;
        DFS(nums, 0);
        return ret;
    }

    void DFS(vector<int>& nums, int pos)
    {
        if(pos == nums.size())
        {
            if(path == _target)
            {
                ret++;
            }

            return;
        }

        // 加
        path += nums[pos];
        DFS(nums, pos + 1);
        path -= nums[pos]; // 回溯,恢复现场

        // 减
        path -= nums[pos];
        DFS(nums, pos + 1);
        path += nums[pos]; // 回溯,恢复现场
    }
};
--------------------------------------------------------------------------
// v2.0,效率有所改善
class Solution 
{
    int ret = 0;
    int _target = 0;
public:
    int findTargetSumWays(vector<int>& nums, int target) 
    {
        _target = target;
        DFS(nums, 0, 0);
        return ret;
    }

    void DFS(vector<int>& nums, int pos, int path)
    {
        if(pos == nums.size())
        {
            if(path == _target)
            {
                ret++;
            }

            return;
        }

        // 加
        DFS(nums, pos + 1, path + nums[pos]);

        // 减
        DFS(nums, pos + 1, path - nums[pos]);
    }
};

3.组合总和

1.题目链接


2.算法原理详解

  • 思路一:每次都只选一个数,此后只能选它及它后面的数

    • 函数设计
      • 全局变量
        • vector<vector<int>> ret
        • vector<int> path
      • DFS()设计:void DFS(nums, pos, sum)
      • 递归出口sum == _target || (sum > _target || pos == nums.size())
      • 回溯 :通过sum控制回溯
      • 剪枝 :控制pos参数,每次从此位置开始递归
  • 思路二:每次枚举一个数,出现几次

    • 函数设计
      • 全局变量
        • vector<vector<int>> ret
        • vector<int> path
      • DFS()设计:void DFS(nums, pos, sum)
      • 递归出口sum == _target || (sum > _target || pos == nums.size())
      • 回溯 :通过sum控制回溯
      • 剪枝 :控制pos参数,每次从此位置开始递归

3.代码实现

cpp 复制代码
// v1.0 每次都只选一个数,此后只能选它及它后面的数
class Solution 
{
    int _target;
    vector<int> path;
    vector<vector<int>> ret;
public:
    vector<vector<int>> combinationSum(vector<int>& nums, int target) 
    {
        _target = target;
        DFS(nums, 0, 0);
        return ret;
    }
    
    void DFS(vector<int>& nums, int pos, int sum)
    {
        if(sum == _target)
        {
            ret.push_back(path);
            return;
        }
        
        if(sum > _target || pos == nums.size())
        {
            return;
        }
        
        // 递归决策 + 剪枝
        for(int i = pos; i < nums.size(); i++)
        {
            path.push_back(nums[i]);
            DFS(nums, i, sum + nums[i]);
            path.pop_back(); // 回溯,恢复现场
        }
    }
};
--------------------------------------------------------------------------
// v2.0 每次枚举一个数,出现几次
class Solution 
{
    int _target;
    vector<int> path;
    vector<vector<int>> ret;
public:
    vector<vector<int>> combinationSum(vector<int>& nums, int target) 
    {
        _target = target;
        DFS(nums, 0, 0);
        return ret;
    }
    
    void DFS(vector<int>& nums, int pos, int sum)
    {
        if(sum == _target)
        {
            ret.push_back(path);
            return;
        }
        
        if(sum > _target || pos == nums.size())
        {
            return;
        }
        
        // 枚举个数 + 剪枝
        for(int i = 0; i * nums[pos] + sum <= _target; i++)
        {
            if(i)
            {
                path.push_back(nums[pos]);    
            }
            
            DFS(nums, pos + 1, i * nums[pos] + sum);
        }
        
        // 回溯,恢复现场
        for(int i = 1; i * nums[pos] + sum <= _target; i++)
        {
            path.pop_back();
        }
    }
};
相关推荐
JingHongB8 分钟前
代码随想录算法训练营Day55 | 图论理论基础、深度优先搜索理论基础、卡玛网 98.所有可达路径、797. 所有可能的路径、广度优先搜索理论基础
算法·深度优先·图论
weixin_4327022611 分钟前
代码随想录算法训练营第五十五天|图论理论基础
数据结构·python·算法·深度优先·图论
小冉在学习13 分钟前
day52 图论章节刷题Part04(110.字符串接龙、105.有向图的完全可达性、106.岛屿的周长 )
算法·深度优先·图论
Repeat71514 分钟前
图论基础--孤岛系列
算法·深度优先·广度优先·图论基础
小冉在学习16 分钟前
day53 图论章节刷题Part05(并查集理论基础、寻找存在的路径)
java·算法·图论
武子康28 分钟前
大数据-212 数据挖掘 机器学习理论 - 无监督学习算法 KMeans 基本原理 簇内误差平方和
大数据·人工智能·学习·算法·机器学习·数据挖掘
passer__jw7671 小时前
【LeetCode】【算法】283. 移动零
数据结构·算法·leetcode
Ocean☾1 小时前
前端基础-html-注册界面
前端·算法·html
顶呱呱程序1 小时前
2-143 基于matlab-GUI的脉冲响应不变法实现音频滤波功能
算法·matlab·音视频·matlab-gui·音频滤波·脉冲响应不变法
Tianyanxiao2 小时前
如何利用探商宝精准营销,抓住行业机遇——以AI技术与大数据推动企业信息精准筛选
大数据·人工智能·科技·数据分析·深度优先·零售