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

力扣题部分:

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;
    }
};


 

  • 11
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
算法训练营主要涵盖了Leetcode题目中的三道题目,分别是Leetcode 28 "Find the Index of the First Occurrence in a String",Leetcode 977 "有序数组的平方",和Leetcode 209 "长度最小的子数组"。 首先是Leetcode 28题,题目要求在给定的字符串中找到第一个出现的字符的索引。思路是使用双指针来遍历字符串,一个指向字符串的开头,另一个指向字符串的结尾。通过比较两个指针所指向的字符是否相等来判断是否找到了第一个出现的字符。具体实现的代码如下: ```python def findIndex(self, s: str) -> int: left = 0 right = len(s) - 1 while left <= right: if s[left == s[right]: return left left += 1 right -= 1 return -1 ``` 接下来是Leetcode 977题,题目要求对给定的有序数组中的元素进行平方,并按照非递减的顺序返回结果。这里由于数组已经是有序的,所以可以使用双指针的方法来解决问题。一个指针指向数组的开头,另一个指针指向数组的末尾。通过比较两个指针所指向的元素的绝对值的大小来确定哪个元素的平方应该放在结果数组的末尾。具体实现的代码如下: ```python def sortedSquares(self, nums: List[int]) -> List[int]: left = 0 right = len(nums) - 1 ans = [] while left <= right: if abs(nums[left]) >= abs(nums[right]): ans.append(nums[left ** 2) left += 1 else: ans.append(nums[right ** 2) right -= 1 return ans[::-1] ``` 最后是Leetcode 209题,题目要求在给定的数组中找到长度最小的子数组,使得子数组的和大于等于给定的目标值。这里可以使用滑动窗口的方法来解决问题。使用两个指针来表示滑动窗口的左边界和右边界,通过移动指针来调整滑动窗口的大小,使得滑动窗口中的元素的和满足题目要求。具体实现的代码如下: ```python def minSubArrayLen(self, target: int, nums: List[int]) -> int: left = 0 right = 0 ans = float('inf') total = 0 while right < len(nums): total += nums[right] while total >= target: ans = min(ans, right - left + 1) total -= nums[left] left += 1 right += 1 return ans if ans != float('inf') else 0 ``` 以上就是第算法训练营的内容。通过这些题目的练习,可以提升对双指针和滑动窗口等算法的理解和应用能力。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值