首页 > 编程语言 >代码随想录算法训练营第二十三天(回溯 二)

代码随想录算法训练营第二十三天(回溯 二)

时间:2024-08-22 21:54:45浏览次数:10  
标签:used target 第二十三 int sum 随想录 vector candidates 回溯

力扣题部分:

39. 组合总和

题目链接:. - 力扣(LeetCode)

题面:

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。

candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。 

对于给定的输入,保证和为 target 的不同组合数少于 150 个。

思路:

记得昨天的组合吗?

这道题和那道题最大的差别就是一个数能不能重复多次选取。

传送门:代码随想录算法训练营第二十二天(回溯 一)-CSDN博客

如果那道题看懂了,这道题唯一的难点就是一个数能重复用代码上是怎么体现的,

可能会有人觉得这一点就会让代码差别巨大,这样的想法其实没什么问题,但是对于这道题,改的其实不多,添一个sum的求和我就不多讲了,下面的代码中最值得注意的还是调用递归那行:

fun(candidates, target, sum, i);

如果你对组合的代码还有印象,那应该记得组合的代码递归时i应该是i + 1才对。

如果递归从i开始,进入递归函数后之前的i变成index,在新的函数中又从index开始,那是不是就可以一直重复了呢?相反,i+1意味着递归的i只能从原来i(index)的下一个开始遍历。

理解完这个点和组合代码的来龙去脉,打出下面的代码就没什么问题了。

代码实现:

class Solution {
public:
    vector<int> rightnums;
    vector<vector<int>> ressult;
    void fun(vector<int> candidates, int target, int sum, int index)
    {
        if(sum == target)
        {
            ressult.push_back(rightnums);
            return;
        }
        if(sum > target) return;
        for(int i = index;i < candidates.size();i ++)
        {
            sum += candidates[i];
            rightnums.push_back(candidates[i]);
            fun(candidates, target, sum, i);
            sum -= candidates[i];
            rightnums.pop_back();
        }
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        fun(candidates,target,0, 0);
        return ressult;
    }
};

40.组合总和II

题目链接:. - 力扣(LeetCode)

题面:

给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用 一次 。

注意:解集不能包含重复的组合。 

思路:

虽然又改动了一些,这次的应对就相对前一道题麻烦不少。

因为数字可以有多个相同的,按照原来的代码,这会造成一个问题,下面我们举个例子:

示例 1:

 candidates = [10,1,2,7,6,1,5], target = 8,

很明显,因为有两个1,我们会得到两个{1,7}。

有人可能会说,将原来数组去掉重复的1不就好了?

如果去掉了两个1中的任何一个,我们又会少掉{1,1,6}这个答案。

看来没什么投机取巧的简单改动了,所以这里选择的是用一个bool类型的数组记录各元素使用情况,不仅如此,为了让同样的元素在一块,我们还要排序,排好之后怎么去重呢?下面的一行代码可以解决去重问题,我先呈现代码,再做解释:

if(i > 0 && candidates[i] == candidates[i - 1] && used[i - 1] == false) continue;

这行代码其实挺难解释的,如果candidates[i] == candidates[i - 1] 并且 used[i - 1] == false,就说明:前一个树枝,使用了candidates[i - 1],也就是说同一树层使用过candidates[i - 1]

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

这块比较抽象,代码随想录弄了一个图解,

这里为了让图弄得不会太复杂,换一个简单一点的例子

candidates = [1, 1, 2], target = 3,(candidates已经排序,相同元素在一起)

我在图中将used的变化用橘黄色标注上,可以看出在candidates[i] == candidates[i - 1]相同的情况下:

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

可能有的录友想,为什么 used[i - 1] == false 就是同一树层呢,因为同一树层,used[i - 1] == false 才能表示,当前取的 candidates[i] 是从 candidates[i - 1] 回溯而来的。

而 used[i - 1] == true,说明是进入下一层递归,去下一个数,所以是树枝上,如图所示:

这块去重的逻辑很抽象,希望大家多理解记忆吧。

代码实现:

class Solution {
public:
    vector<int> rightnums;
    vector<vector<int>> result;
    void fun(vector<int>candidates, int target, int sum, int index, vector<bool> &used)
    {
        if(sum > target) return;
        if(sum == target)
        {

            result.push_back(rightnums);
            return;
        }
        for(int i = index; i < candidates.size(); i ++)
        {
            if(i > 0 && candidates[i] == candidates[i - 1] && used[i - 1] == false) continue;
            sum += candidates[i];
            rightnums.push_back(candidates[i]);
            used[i] = true;
            fun(candidates, target, sum, i + 1, used);
            rightnums.pop_back();
            sum -= candidates[i];
            used[i] = false;
        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<bool> used(candidates.size(), false);
        sort(candidates.begin(), candidates.end());
        fun(candidates,target, 0, 0, used);
        return result;
    }
};

131.分割回文串

题目链接:. - 力扣(LeetCode)

题面:

给你一个字符串 s,请你将 s 分割成一些子串,使每个子串都是 回文串

回文串 是向前和向后读都相同的字符串。)

 返回 s 所有可能的分割方案。

思路:

本题这涉及到两个关键问题:

  1. 切割问题,有不同的切割方式
  2. 判断回文

这种题目,想用for循环暴力解法,可能都不那么容易写出来,所以要换一种暴力的方式,就是回溯。

一些同学可能想不清楚 回溯究竟是如何切割字符串呢?

我们来分析一下切割,其实切割问题类似组合问题

例如对于字符串abcdef:

  • 组合问题:选取一个a之后,在bcdef中再去选取第二个,选取b之后在cdef中再选取第三个.....。
  • 切割问题:切割一个a之后,在bcdef中再去切割第二段,切割b之后在cdef中再切割第三段.....。

所以切割问题,也可以抽象为一棵树形结构,如图:

131.分割回文串

递归用来纵向遍历,for循环用来横向遍历,切割线(就是图中的红线)切割到字符串的结尾位置,说明找到了一个切割方法。

那么在代码里什么是切割线呢?

在处理组合问题的时候,递归参数需要传入startIndex,表示下一轮递归遍历的起始位置,这个startIndex就是切割线,可以结合下面的判断终止条件的代码理解一下

void backtracking (const string& s, int startIndex) {
    // 如果起始位置已经大于s的大小,说明已经找到了一组分割方案了
    if (startIndex >= s.size()) {
        result.push_back(path);
        return;
    }
}

终止条件想明白了的话其实难点也差不多没了,剩下的就是for循环遍历和判断回文了,个人感觉没什么其他太多值得说的,我就直接呈现套模板的代码了。

代码实现:

class Solution {
public:
    vector<vector<string>> result;
    vector<string> right; // 放已经回文的子串
    bool judge(const string &s, int begin, int end)
    {
        int i, j;
        for(i = begin, j = end; i < j; i ++, j --)
        {
            if(s[i] != s[j]) return false;
        }
        return true;
    }
    void backtracking (const string &s, int startIndex) 
    {
        if(startIndex >= s.size())
        {
            result.push_back(right);
            return;
        }
        for(int i = startIndex; i < s.size(); i ++)
        {
            if(judge(s, startIndex, i))
            {
                string son = s.substr(startIndex,i - startIndex + 1);
                right.push_back(son);
                backtracking(s, i + 1);
                right.pop_back();
            }
        }
    }
    vector<vector<string>> partition(string s) 
    {
        backtracking(s, 0);
        return result;
    }
};


 

标签:used,target,第二十三,int,sum,随想录,vector,candidates,回溯
From: https://blog.csdn.net/b1ankwall/article/details/141431005

相关文章

  • 代码随想录算法训练营第二十二天(回溯 一)
    开始学习回溯!回溯理论基础代码随想录文章链接:代码随想录文章摘要:什么是回溯法回溯法也可以叫做回溯搜索法,它是一种搜索的方式。在二叉树系列中,我们已经不止一次,提到了回溯。回溯是递归的副产品,只要有递归就会有回溯。所以以下讲解中,回溯函数也就是递归函数,指的都是一......
  • 代码随想录Day23
    131.分割回文串给你一个字符串s,请你将s分割成一些子串,使每个子串都是回文串。返回s所有可能的分割方案。示例1:输入:s="aab"输出:[["a","a","b"],["aa","b"]]示例2:输入:s="a"输出:[["a"]]提示:1<=s.length<=16s仅由......
  • 「代码随想录算法训练营」第四十四天 | 图论 part2
    200.岛屿数量题目链接:https://leetcode.cn/problems/number-of-islands/description/文章讲解:https://programmercarl.com/kamacoder/0099.岛屿的数量深搜.html题目难度:中等题目状态:看题解思路一:深搜版方法dfs:参数:接受一个字符网格grid和当前坐标(r,c)。功能:......
  • 给自己复盘的随想录笔记-移除元素
    双指针法双指针法(快慢指针法): 通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。定义快慢指针快指针:寻找新数组的元素,新数组就是不含有目标元素的数组慢指针:指向更新新数组下标的位置相关题目删除有序数组中的重复项解题思路:解法:双指针首先注意数组......
  • 给自己复盘用的随想录笔记day1-二分查找
    二分法使用情景数组为有序数组,同时题目还强调数组中无重复元素,因为一旦有重复元素,使用二分查找法返回的元素下标可能不是唯一的,这些都是使用二分法的前提条件,当大家看到题目描述满足如上条件的时候,可要想一想是不是可以用二分法了。求某一个整数的平方根边界条件写二分法......
  • 常用算法 -- 回溯算法
    1简介        回溯算法是一种基于试错思想的搜索算法,也是一种重要的编程技巧,常用于解决组合、排列、切割等问题。它通过不断尝试解决问题的每一步,一旦发现当前步骤不能得到有效的正确解,就会返回上一步或多步,并尝试其他可能的分步解决方案。这种策略使得回溯算法能够......
  • 代码随想录 -- 数组 -- 螺旋矩阵II
    59.螺旋矩阵II-力扣(LeetCode)每画一条边都要坚持一致的左闭右开注意处理n为奇数时的矩阵中心点classSolution(object):defgenerateMatrix(self,n):res=[[0]*nforainrange(n)]startX=0startY=0loop=mid=n/2c......
  • 代码随想录 -- 数组 -- 区间和
    58.区间和(第九期模拟笔试)(kamacoder.com)暴力解法大概率超时,应采用前缀和解法p[i] 表示vec[0]到vec[i]的累加和求vec[m] 到vec[n] 的和只需要 p[n]-p[m] 即可知识点input函数Python3 中raw_input()和input()进行了整合,去除了raw_input(),仅保留了i......
  • 代码随想录day37 || 518 零钱兑换,377 组合总和iv,70 爬楼梯
    0-1背包问题在0-1背包问题中,每种物品只能选择一次,因此一旦选择某个物品后,剩余的容量只能放入前面的物品。这就是为什么状态转移方程是:dp[i][j]=max(dp[i-1][j],dp[i-1][j-w(i)]+v(i))这里的dp[i-1][j-w(i)]+v(i)表示选择第(i)个物品后,剩余的容量只能放入前(......
  • 代码随想录Day22
    77.组合给定两个整数n和k,返回范围[1,n]中所有可能的k个数的组合。你可以按任何顺序返回答案。示例1:输入:n=4,k=2输出:[[2,4],[3,4],[2,3],[1,2],[1,3],[1,4],]示例2:输入:n=1,k=1输出:[[1]]提示:1<=n<=201<=k<=n正解......