代码随想录第22天： 77. 组合、216.组合总和III、17.电话号码的字母组合

 77. 组合，应用回溯解决组合问题的经典模板，代码如下：

// 第77题. 组合,回溯问题

#include <vector>
#include <iostream>
using namespace std;

vector<vector<int>> result;
vector<int> path;

void backtracking(int n, int k, int startindex)
{
    if (path.size() == k) // 当path满足大小时将结果放回，并直接返回
    {
        result.push_back(path);
        return;
    }
    for (int i = startindex; i <= n; i++) // for循环控制第一层树选择有几种可能
    {
        path.push_back(i);
        backtracking(n, k, i + 1);
        path.pop_back(); // 当收集一个结果时不会执行for循环，所以层内收集一个弹出一个，涉及层切换时会弹出k次, 比较不容易理解的地方
    }
}

vector<vector<int>> combine(int n, int k)
{
    result.clear(); // 可以不写
    path.clear();   // 可以不写
    backtracking(n, k, 1);
    return result;
}

int main()
{
    int n = 4;
    int k = 2;
    vector<vector<int>> res = combine(n, k);
    for (vector<int> a : res)
    {
        cout << "[";
        for (int b : a)
        {
            b == a[0] ? cout << b << "," : cout << b;
        }
        cout << "]";
    }
}

216.组合总和III，组合类型的变种，只需要判断组合的和即可

// 第77题. 组合,回溯问题

#include <vector>
#include <iostream>
using namespace std;

vector<vector<int>> result;
vector<int> path;

void backtracking(int targetsum, int k, int startindex, int sum)
{
    if (path.size() == k) // 当path满足大小时将结果放回，并直接返回
    {
        if (sum == targetsum)
            result.push_back(path);
        return;
    }
    for (int i = startindex; i <= 9; i++) // for循环控制第一层树选择有几种可能
    {
        sum += i;
        path.push_back(i);
        backtracking(targetsum, k, i + 1, sum);
        sum -= i;
        path.pop_back(); // 当收集一个结果时不会执行for循环，所以层内收集一个弹出一个，涉及层切换时会弹出k次, 比较不容易理解的地方
    }
}

vector<vector<int>> combineSum3(int k, int n)
{
    result.clear(); // 可以不写
    path.clear();   // 可以不写
    backtracking(n, k, 1, 0);
    return result;
}

int main()
{
    int n = 7;
    int k = 3;
    vector<vector<int>> res = combineSum3(k, n);
    for (vector<int> a : res)
    {
        cout << "[";
        for (int b : a)
        {
            b == a[k - 1] ? cout << b : cout << b << ",";
        }
        cout << "]";
    }
    return 0;
}

17.电话号码的字母组合 ，同样套用组合回溯模板，注意使用字符串实现数字和字母映射关系的细节

// 17.电话号码的字母组合

#include <vector>
#include <iostream>
#include <string>

using namespace std;

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;
string s;
void backtracking(const string &digits, int index)
{
    if (index == digits.size())
    {
        result.push_back(s);
        return;
    }
    int digit = digits[index] - '0';   // 将index指向的数字转为int
    string letters = letterMap[digit]; // 取数字对应的字符集
    for (int i = 0; i < letters.size(); i++)
    {
        s.push_back(letters[i]);         // 处理
        backtracking(digits, index + 1); // 递归，注意index+1，一下层要处理下一个数字了
        s.pop_back();                    // 回溯
    }
}

vector<string> letterCombinations(string digits)
{
    s.clear();
    result.clear();
    if (digits.size() == 0)
    {
        return result;
    }
    backtracking(digits, 0);
    return result;
}

int main()
{
    string st = "23";
    vector<string> res = letterCombinations(st);
    for (string a : res)
    {
        cout << a << endl;
    }
    return 0;
}