24暑假算法刷题 | Day23 | LeetCode 39. 组合总和,40. 组合总和 II,131. 分割回文串


39. 组合总和

点此跳转题目链接

题目描述

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

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

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

示例 1:

输入:candidates = [2,3,6,7], target = 7
输出:[[2,2,3],[7]]
解释:
2 和 3 可以形成一组候选,2 + 2 + 3 = 7 。注意 2 可以使用多次。
7 也是一个候选, 7 = 7 。
仅有这两种组合。

示例 2:

输入: candidates = [2,3,5], target = 8
输出: [[2,2,2,2],[2,3,3],[3,5]]

示例 3:

输入: candidates = [2], target = 1
输出: []

提示:

  • 1 <= candidates.length <= 30
  • 2 <= candidates[i] <= 40
  • candidates 的所有元素 互不相同
  • 1 <= target <= 40

题解

经典的回溯算法题目,和 打卡-回溯算法 I 中的几道题的差不多,直接按照 回溯三部曲——处理、递归、回溯 来解决即可:

class Solution
{
private:
    vector<int> path;
    vector<vector<int>> res;

public:
    void backTracking(const vector<int> &candidates, int target, int curSum, int start)
    {
        // 递归出口(纵向遍历)
        if (curSum == target)
        {
            res.push_back(path);
            return;
        }

        // 剪枝
        if (curSum > target)
            return;

        // 横向遍历
        for (int i = start; i < candidates.size(); ++i)
        {
            path.push_back(candidates[i]);
            curSum += candidates[i];                     // 处理
            backTracking(candidates, target, curSum, i); // 递归
            path.pop_back();                             // 回溯
            curSum -= candidates[i];
        }
    }

    vector<vector<int>> combinationSum(vector<int> &candidates, int target)
    {
        backTracking(candidates, target, 0, 0);
        return res;
    }
};

注意一下题目说“ candidates 中的 同一个 数字可以 无限制重复被选取 ”,所以递归时:

backTracking(candidates, target, curSum, i); // 递归

最后一个参数传的还是 i ,而不是许多类似题目中的 i + 1


40. 组合总和 II

点此跳转题目链接

题目描述

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

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

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

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]

示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
输出:
[
[1,2,2],
[5]
]

提示:

  • 1 <= candidates.length <= 100
  • 1 <= candidates[i] <= 50
  • 1 <= target <= 30

题解

这题的基本框架还是很经典的回溯算法,和 打卡-回溯算法 I 中的几道题差不多,按部就班地写出 回溯三部曲

  • 处理
  • 递归
  • 回溯

就能得到底层代码:

class Solution
{
private:
    vector<int> path;
    vector<vector<int>> res;

public:
    void backTracking(const vector<int> &candidates, int target, int curSum, int start)
    {
        // 递归出口(纵向遍历)
        if (curSum == target)
        {
            res.push_back(path);
            return;
        }

        // 剪枝
        if (curSum > target)
            return;

        // 横向遍历
        for (int i = start; i < candidates.size(); ++i)
        {
            path.push_back(candidates[i]);
            curSum += candidates[i];                         // 处理
            backTracking(candidates, target, curSum, i + 1); // 递归
            path.pop_back();                                 // 回溯
            curSum -= candidates[i];
        }
    }

    vector<vector<int>> combinationSum2(vector<int> &candidates, int target)
    {
        backTracking(candidates, target, 0, 0);
        return res;
    }
};

但是这样会出现 重复 的问题,例如:

在这里插入图片描述

于是下一个难点就是 去重 了。首先尝试了简单地用 set / map 之类的全局记录去重,但是这样容易在一些阴间测试用例下超时:

在这里插入图片描述

也就是说,我们还是需要在 搜索过程中 就完成去重。

引入一个用于去重的数组 used ,大小和 candidates 相同,表示 candidates 中对应位置的数是否在当前的组合中已经使用过。根据题目描述, candidates 中的每个数字在每个组合中只能使用一次,且解集中不能包含重复的组合,也就是说,在 回溯的树形结构图 中, 同一个树枝上 可以有重复,但是某节点下的 同一层 不能有重复:

⚠️ 这种算法需要先将 candidates 排序 ,目的是将重复的数字放在一起,便于去重

在这里插入图片描述

图片来源:代码随想录-40

这里判断重复的方法是:

如果 candidates[i] == candidates[i - 1] used[i- 1] == 0 ,则说明 同一层的前一个节点已使用过 candidates[i - 1] 这个与当前数字重复的值 ,需要去重(即跳过当前的 candidates[i] )。

上面的两个条件中,前者好理解(就是遇到重复数字),后者的原因在于:

  • 如果 used[i - 1] == 1 ,说明 candidates[i - 1] 之前加到了当前组合中,也就是这是 同一个树枝 上的操作,即通过 递归纵向遍历 得到的

  • 如果 used[i - 1] == 0 ,说明 candidates[i - 1] 不在当前组合中,那么它应该是在 树中同一层的前一个节点 使用过了,即通过 回溯横向遍历 得到的

下面搬运Carl的两张图便于进一步理解:

在这里插入图片描述

在这里插入图片描述

代码(C++)

class Solution
{
private:
    vector<int> path;
    vector<vector<int>> res;
    vector<int> used;

public:
    void backTracking(const vector<int> &candidates, int target, int curSum, int start)
    {
        // 递归出口(纵向遍历)
        if (curSum == target)
        {
            res.push_back(path);
            return;
        }

        // 剪枝
        if (curSum > target)
            return;

        // 横向遍历
        for (int i = start; i < candidates.size(); ++i)
        {
            // 去重
            if (i > start && candidates[i] == candidates[i - 1] && !used[i - 1])
                continue;
            // 处理
            path.push_back(candidates[i]);
            curSum += candidates[i];       
            used[i] = 1;
            // 递归
            backTracking(candidates, target, curSum, i + 1); 
            // 回溯
            path.pop_back();                                 
            curSum -= candidates[i];
            used[i] = 0;
        }
    }

    vector<vector<int>> combinationSum2(vector<int> &candidates, int target)
    {
        used.resize(candidates.size());
        sort(candidates.begin(), candidates.end()); // 先排序,便于搜索过程中去重
        backTracking(candidates, target, 0, 0);
        return res;
    }
};

131. 分割回文串

点此跳转题目链接

题目描述

给你一个字符串 s,请你将 s 分割成一些子串,使每个子串都是 回文串 。返回 s 所有可能的分割方案。

回文串:向前和向后读都相同的字符串

示例 1:

输入:s = "aab"
输出:[["a","a","b"],["aa","b"]]

示例 2:

输入:s = "a"
输出:[["a"]]

提示:

  • 1 <= s.length <= 16
  • s 仅由小写英文字母组成

题解

还是经典的回溯算法,和基础的 组合问题 略有不同:

组合问题 是每次“将新的元素加入当前组合”,递归出口一般是“当前组合长度达到目标长度”或者其他特殊条件;本题的 分割问题 思路是每次“尝试在下一个位置画分割线(如果新分割截断的子串是回文的就保留)”,递归出口为“分割线位置到达字符串末尾位置”。

整体框架仍是 回溯三部曲——处理、递归、回溯

class Solution
{
private:
    vector<string> path;
    vector<vector<string>> res;

    // 判断是否为回文串
    bool isPalindrome(const string &str)
    {
        for (int i = 0, j = str.size() - 1; i <= j; ++i, --j)
        {
            if (str[i] != str[j])
                return false;
        }
        return true;
    }

public:
    void backTracking(const string &s, int cutPos)
    {
        // 递归出口:切割点到达字符串末尾位置
        if (cutPos >= s.size())
        {
            res.push_back(path);
            return;
        }
        // 横向遍历
        for (int i = cutPos; i < s.size(); ++i)
        {
            // 处理:判断截断的子串是不是回文串
            string sub = s.substr(cutPos, i - cutPos + 1);
            if (isPalindrome(sub))
                path.push_back(sub);
            else
                continue;
            backTracking(s, i + 1); // 递归
            path.pop_back();        // 回溯
        }
    }

    vector<vector<string>> partition(string s)
    {
        backTracking(s, 0);
        return res;
    }
};
  • 31
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值