【回溯算法篇】组合问题(下)

简介: 【回溯算法篇】组合问题(下)

👉组合总和 III👈


d43004d410914ae6bd85bed0c7c05528.png

其实组合总和 III 的解题思路也和组合的思路是一样的。也是套用回溯算法的模板,就是修改一下递归终止的条件和 for 循环横向遍历是做到操作。


af95aa48198e4feda4865a37ca03c8b6.png


class Solution 
{
private:
    vector<vector<int>> result;
    vector<int> path;
    // tagetSum为目标和
    // k为要求每个组合中数字的个数
    // startIndex为下一层递归开始遍历的下标
    // sum为路径和
    void backtracking(int targetSum, int k, int startIndex, int sum)
    {
        if(path.size() == k)
        {
            if(sum == targetSum)
            {
                result.push_back(path); // 收集结果
            }
            return;
        }
        // 横向遍历
        for(int i = startIndex; i <= 9; ++i)
        {
            // 处理节点
            sum += i;
            path.push_back(i);
            // 递归,纵向遍历
            backtracking(targetSum, k, i + 1, sum);
            // 回溯
            sum -= i;
            path.pop_back();
        }
    }
public:
    vector<vector<int>> combinationSum3(int k, int n) 
    {
        result.clear(); // 可以不做
        path.clear();   // 可以不做
        backtracking(n, k, 1, 0);
        return result;
    }
};


注:startIndex 是为了防止重复选数的。


组合总和 III 的剪枝优化有两个地方可以优化,第一个是和组合的剪枝优化一下,就是已经选的元素个数加上剩余的元素个数无法到达 k;还有一个优化就是,如果 sum 已经大于 targetSum 了,就没必要再往下搜索了(尽管所需要元素个数没有达到 k 个)。这个优化当做递归的终止条件,也可以放在 for 横向遍历中。下方的代码是放在 for 的横向遍历中,注意剪枝之前需要把回溯做了。


class Solution 
{
private:
    vector<vector<int>> result;
    vector<int> path;
    // tagetSum为目标和
    // k为要求每个组合中数字的个数
    // startIndex为下一层递归开始遍历的下标
    // sum为路径和
    void backtracking(int targetSum, int k, int startIndex, int sum)
    {        
        if(path.size() == k)
        {
            if(sum == targetSum)
            {
                result.push_back(path);
            }
            return;
        }
                                    // 剪枝优化
        for(int i = startIndex; i <= 9 + 1 - (k - path.size()); ++i)
        {
            sum += i;
            path.push_back(i);
            if(sum > targetSum) // 剪枝优化
            {
                sum -= i;   // 剪枝之前先把回溯做了
                path.pop_back();    // 剪枝之前先把回溯做了
                return;
            }
            backtracking(targetSum, k, i + 1, sum);
            sum -= i;
            path.pop_back();
        }
    }
public:
    vector<vector<int>> combinationSum3(int k, int n) 
    {
        backtracking(n, k, 1, 0);
        return result;
    }
};

2ef5331442314be1b5abe817713f336d.png


👉组合总和👈


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


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


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

2437d6f22c0d40478151918608a3e975.png


组合总和这道题目是给定了候选数字的数组和目标和。那么 for 循环遍历的集合就变成了候选数字数组。因为可以选取重复的元素,那么下一层递归还是从 i 开始搜索。注:因为给出的数字集合有序的,所以可以先将其排序。


class Solution 
{
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(const vector<int>& candidates, int target, int sum, int startIndex)
    {
        // 收集结果
        if(sum == target)
        {
            result.push_back(path);
            return;
        }
                                                // 剪枝优化
        for(int i = startIndex; i < candidates.size() && sum + candidates[i] <= target; ++i)
        {
            // 各种可能
            path.push_back(candidates[i]);
            sum += candidates[i];
            //不用i+1了,表示可以重复读取当前的数
            backtracking(candidates, target, sum, i);
            // 回溯
            sum -= candidates[i];
            path.pop_back();
        }
    }
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) 
    {
        sort(candidates.begin(), candidates.end());
        backtracking(candidates, target, 0, 0);
        return result;
    }
};

8f057b7e56904a86819f4707803c6867.png



272d2c3dd5b34f698a324d5db2cf3c57.png


👉组合总和 II👈


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


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


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

3e04eb9456ae42048c596407f11b9d56.png



本题数组 candidates 的元素是有重复的,而组合总和是无重复元素的数组 candidates。本道题的难点就是数组 candidates 有重复的元素,但不能有重复的组合,所以我们要在搜索的过程中去掉重复的组合。


所以的去重,就是使用过的元素不能重复选取。因为组合问题可以抽象成树形结构,那么使用过在树形结构上是有两个维度的,一个维度是同一树枝上使用过,一个维度是同一树层上使用过。那么问题来了,我们所要的结果是同一树枝上使用过的还是同一树层上使用过的。


看一下题目的输出结果,一个组合中是可以有相同元素的,但是两个组合是不能够相同的。那么我们要去重的就是同一树层上使用过的,同一树枝上的都是一个组合里的元素,不需要去重。那么我们需要加一个 bool 类型数组 used,用来记录同一树枝上的元素是否使用过。


3c6f2fe39fce44e5a0ce9c0242f8f240.png

当 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,说明是进入下一层递归,去下一个数,所以是树枝上,如图所示:


1708e1eaa7a9492cbdb61143d033a509.png

class Solution 
{
private:
    vector<vector<int>> ret;
    vector<int> path;
    void BackTracking(const vector<int>& candidates, int target, int sum, int startIndex, vector<bool> used)
    {
        if(sum == target)
        {
            ret.push_back(path);
            return;
        }
                                            // 剪枝优化
        for(size_t i = startIndex; i < candidates.size() && sum + candidates[i] <= target; ++i)
        {
            // used[i - 1] == true,说明同一树枝candidates[i - 1]使用过
            // used[i - 1] == false,说明同一树层candidates[i - 1]使用过
            // 要对同一树层使用过的元素进行跳过
            if(i > 0 && candidates[i] == candidates[i - 1] && used[i - 1] == false)
                continue;
            used[i] = true;
            path.push_back(candidates[i]);
            sum += candidates[i];
            // i+1表示一个数不能被重复选取
            BackTracking(candidates, target, sum, i + 1, used); 
            // 回溯
            used[i] = false;
            path.pop_back();
            sum -= candidates[i];
        }
    }
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) 
    {
        ret.clear();
        path.clear();
        size_t n = candidates.size();
        // 首先把给candidates排序,让其相同的元素都挨在一起。
        sort(candidates.begin(), candidates.end());
        vector<bool> used(n, false);
        BackTracking(candidates, target, 0, 0, used); 
        return ret;
    }
};

df59356eaab84a1d931631159c842148.png

注:这直接用 startIndex 来去重也是可以的, 就不用used 数组了。


class Solution 
{
private:
    vector<vector<int>> ret;
    vector<int> path;
    void BackTracking(const vector<int>& candidates, int target, int sum, int startIndex)
    {
        if(sum == target)
        {
            ret.push_back(path);
            return;
        }
                                            // 剪枝优化
        for(size_t i = startIndex; i < candidates.size() && sum + candidates[i] <= target; ++i)
        {
            // 要对同一树层使用过的元素进行跳过
            if(i > startIndex && candidates[i] == candidates[i - 1])
                continue;
            path.push_back(candidates[i]);
            sum += candidates[i];
            // i+1表示一个数不能被重复选取
            BackTracking(candidates, target, sum, i + 1); 
            // 回溯
            path.pop_back();
            sum -= candidates[i];
        }
    }
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) 
    {
        ret.clear();
        path.clear();
        // 首先把给candidates排序,让其相同的元素都挨在一起。
        sort(candidates.begin(), candidates.end());
        BackTracking(candidates, target, 0, 0); 
        return ret;
    }
};


1e5b40d0058541d98727cae90d91dbb8.png


👉电话号码的字母组合👈


给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。答案可以按任意顺序返回。


给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。

7cb6d26eb2d64c49ab0abb98475b2d6e.png

c0ff639797284c818ae2e87fcfee3759.png



b3801bd7e8574f4b84d5d6dad925f7ac.png

 

这道题是比较简单的,和上面的题的思路差不多,就不进行过多的讲解了,下面的代码有相应的注释。


class Solution 
{
private:
    // 数字和字符串之间的映射可以采用哈希表
    const string letterMap[10] = { "", "", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz" };
    vector<string> ret;
    string str;
    void BackTracking(const string& digits, int startIndex)
    {
        // 当下标和digits.size()是说明可以收集结果了
        if(startIndex == digits.size())
        {
            ret.push_back(str);
            return;
        }
        int digit = digits[startIndex] - '0';   // 将字符'1'指向的数字1
        string letters = letterMap[digit];  // 拿到数字对应的字符串
        for(size_t i = 0; i < letters.size(); ++i)
        {
            str.push_back(letters[i]);
            //递归,startIndex+1表明一下层要处理下一个数字了
            BackTracking(digits, startIndex + 1);  
            str.pop_back();
        }
    }
public:
    vector<string> letterCombinations(string digits) 
    {
        ret.clear();
        str.clear();
        if(digits.empty())
            return ret;
        BackTracking(digits, 0);
        return ret;
    }
};

e92d0fe905664771a202af3d1d6f71ec.png


class Solution {
private:
    // letterMap是数字到字母串的映射
    const string letterMap[10] = 
    {
        "",      // 0 
        "",      // 1
        "abc",   // 2
        "def",   // 3
        "ghi",   // 4
        "jkl",   // 5
        "mno",   // 6
        "pqrs",  // 7
        "tuv",   // 8
        "wxyz"   // 9
    };
    vector<string> result;
    // index为递归的深度,也为digits数字串的下标
    void backtracking(const string& digits, int index, const string& s)
    {
        // 收集结果
        if(index == digits.size())
        {
            result.push_back(s);
            return;
        }
        int digit = digits[index] - '0';
        string letters = letterMap[digit];  // letters是数字对应的字母串
        for(int i = 0; i < letters.size(); ++i)
        {
            backtracking(digits, index + 1, s + letters[i]);   // 回到上一层是还是s
        }
    }
public:
    vector<string> letterCombinations(string digits) 
    {
        if(digits.size() == 0)
        {
            return result;
        }
        backtracking(digits, 0, "");
        return result;
    }
};

b5cda111f3044b9ca72257f055740752.png


👉总结👈


本篇博客主要讲解了什么是回溯算法、回溯算法主要解决的问题、回溯算法的模板、组合问题以及剪枝优化等。那么以上就是本篇博客的全部内容了,如果大家觉得有收获的话,可以点个三连支持一下!谢谢大家!💖💝❣️





相关文章
|
17天前
|
存储 算法 安全
2024重生之回溯数据结构与算法系列学习之串(12)【无论是王道考研人还是IKUN都能包会的;不然别给我家鸽鸽丟脸好嘛?】
数据结构与算法系列学习之串的定义和基本操作、串的储存结构、基本操作的实现、朴素模式匹配算法、KMP算法等代码举例及图解说明;【含常见的报错问题及其对应的解决方法】你个小黑子;这都学不会;能不能不要给我家鸽鸽丢脸啊~除了会黑我家鸽鸽还会干嘛?!!!
2024重生之回溯数据结构与算法系列学习之串(12)【无论是王道考研人还是IKUN都能包会的;不然别给我家鸽鸽丟脸好嘛?】
|
17天前
|
算法 安全 搜索推荐
2024重生之回溯数据结构与算法系列学习(8)【无论是王道考研人还是IKUN都能包会的;不然别给我家鸽鸽丢脸好嘛?】
数据结构王道第2.3章之IKUN和I原达人之数据结构与算法系列学习x单双链表精题详解、数据结构、C++、排序算法、java、动态规划你个小黑子;这都学不会;能不能不要给我家鸽鸽丢脸啊~除了会黑我家鸽鸽还会干嘛?!!!
|
17天前
|
存储 算法 安全
2024重生之回溯数据结构与算法系列学习之顺序表【无论是王道考研人还真爱粉都能包会的;不然别给我家鸽鸽丢脸好嘛?】
顺序表的定义和基本操作之插入;删除;按值查找;按位查找等具体详解步骤以及举例说明
|
17天前
|
算法 安全 搜索推荐
2024重生之回溯数据结构与算法系列学习之单双链表精题详解(9)【无论是王道考研人还是IKUN都能包会的;不然别给我家鸽鸽丢脸好嘛?】
数据结构王道第2.3章之IKUN和I原达人之数据结构与算法系列学习x单双链表精题详解、数据结构、C++、排序算法、java、动态规划你个小黑子;这都学不会;能不能不要给我家鸽鸽丢脸啊~除了会黑我家鸽鸽还会干嘛?!!!
|
17天前
|
存储 Web App开发 算法
2024重生之回溯数据结构与算法系列学习之单双链表【无论是王道考研人还是IKUN都能包会的;不然别给我家鸽鸽丢脸好嘛?】
数据结构之单双链表按位、值查找;[前后]插入;删除指定节点;求表长、静态链表等代码及具体思路详解步骤;举例说明、注意点及常见报错问题所对应的解决方法
|
17天前
|
算法 安全 NoSQL
2024重生之回溯数据结构与算法系列学习之栈和队列精题汇总(10)【无论是王道考研人还是IKUN都能包会的;不然别给我家鸽鸽丢脸好嘛?】
数据结构王道第3章之IKUN和I原达人之数据结构与算法系列学习栈与队列精题详解、数据结构、C++、排序算法、java、动态规划你个小黑子;这都学不会;能不能不要给我家鸽鸽丢脸啊~除了会黑我家鸽鸽还会干嘛?!!!
|
17天前
|
算法 安全 NoSQL
2024重生之回溯数据结构与算法系列学习之顺序表习题精讲【无论是王道考研人还真爱粉都能包会的;不然别给我家鸽鸽丢脸好嘛?】
顺序表的定义和基本操作之插入;删除;按值查找;按位查找习题精讲等具体详解步骤以及举例说明
|
17天前
|
存储 算法 安全
2024重生之回溯数据结构与算法系列学习【无论是王道考研人还真爱粉都能包会的;不然别给我家鸽鸽丢脸好嘛?】
数据结构的基本概念;算法的基本概念、特性以及时间复杂度、空间复杂度等举例说明;【含常见的报错问题及其对应的解决方法】
|
17天前
|
算法 安全 搜索推荐
2024重生之回溯数据结构与算法系列学习之王道第2.3章节之线性表精题汇总二(5)【无论是王道考研人还是IKUN都能包会的;不然别给我家鸽鸽丢脸好嘛?】
IKU达人之数据结构与算法系列学习×单双链表精题详解、数据结构、C++、排序算法、java 、动态规划 你个小黑子;这都学不会;能不能不要给我家鸽鸽丢脸啊~除了会黑我家鸽鸽还会干嘛?!!!
|
20天前
|
算法
”回溯算法“框架及练习题
”回溯算法“框架及练习题
30 0