代码随想录算法训练营第二十五天| 216 组合总和III 17 电话号码的字母组合

代码随想录算法训练营第二十五天| 216 组合总和III 17 电话号码的字母组合

LeetCode 216 组合总和

题目链接: 216.组合总和III

思路：和leetcode 77.组合相似，只是添加了一个和的限制

回溯三部曲

• 回溯函数模板返回值及参数

本题中仍然需要全局变量result和path

vector<vector<int> > result;
vector<int> path;
void backtracking(int targetSum, int k, int sum, int startIndex)

• 回溯终止条件

此题终止条件和上题一样，取到k个元素即可

if(path.size() == k) {
    if(sum == targetSum) result.push_back(path);  //将符合条件的数组放入result
    return;
}

• 回溯搜索遍历过程

本题集合固定为[1,…,9]，for循环固定i<=9，然后path收集每次选取元素，sum统计path里面总和

for(int i = startIndex; i <= 9; i++) {
    sum += i;
    path.push_back(i);
    backtracking(targetSum, k, sum, i + 1);
    sum -= i;  //回溯
    path.pop_back();
}

完整代码：

class Solution {
private:
    vector<vector<int> > result;
    vector<int> path;
    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) {
        backtracking(n, k, 1, 0);
        return result;
    }
};

减枝操作

class Solution{
private:
    vector<vector<int> > result;
    vector<int> path;
    void backtracking(int targetSum, int k, int startIndex, int sum) {
        if(sum > targetSum) return;
        if(path.size() == k) {
            if(sum == targetSum) result.push_back(path);
            return;
        }
        for(int i = startIndex; i <= 9 - (k - path.size()) + 1; 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) {
        backtracking(n, k, 1, 0);
        return result;
    }
};

LeetCode 17 电话号码的字母组合

题目链接: 17.电话号码的字母组合

思路：如果首先想到暴力法，如遇到"23"尝试使用两层for循环进行遍历，但是如果"233"，"23333…"则无法进行遍历，此时考虑回溯法

根据题意，需要解决如下几个问题：

• 数字和字母如何完成映射
• 1*#按键异常情况如何处理

可以使用map或定义二维数组，如：string letterMap[10]来做映射

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)  //此处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);  //递归
    s.pop_back();  //回溯
}

整体代码为：

class Solution{
private:
    const string letterMap[10] = {
        "",  //0
        "",  //1
        "abc",  //2
        "def",  //3
        "ghi",  //4
        "jkl",  //5
        "mno",  //6
        "pqrs",  //7
        "tuv",  //8
        "wxyz"  //9
    };
public:
    vector<string> result;
    string s;
    void backtracking(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(leters[i]);
            backtracking(digits, index + 1);
            s.pop_back();  //回溯
        }
    }
    vector<string> letterCombinations(string digits) {
        if(digits.size() == 0) return result;
        backtracking(digits, 0);
        return result;
    }
};