22.回溯(2) | 组合总和III、电话号码的字母组合

        今天的2道题是不同风格的,第1道是取或不取当前元素,或下一个元素取第几个的问题。是在同一个集合中选取元素进行组合,核心在于横向与纵向剪枝;第2道是当前下标下一定会取元素,范围也确定,不确定取哪个的问题。是在多个集合中选取元素进行组合,核心在于用字母表转化问题。整体上都比较容易。


        第1题(216. 组合总和III)相比day 24中的模板只是对组合中数字的和限定为了n,所以在递归出口处判断和是否为n即可。记录和的具体方法可以像之前day 18中第2题(112. 路径总和)一样转化为减法,判断最终结果是否为0。这道题按上面的写法虽然也能AC,但还可以优化,重点在于剪枝,横向和纵向都可进行剪枝:

  • 纵向上,因为要求k个数,如果在目前存的数字个数的基础上,对之后的所有数字都保存也达不到k个,那么就没有进行尝试的必要了。所以只有在i <= 9 - (k - path.size()) + 1时才进行尝试(此时path.size()一定小于k,因为否则在前面就进入递归出口了)。
  • 横向上,如果当前路径上数字和已经 ≥ n,也就没有尝试的必要了。所以只有在n - i >= 0时才进行尝试。
class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;
    void back(int sum, int k, int begin) {
        if (path.size() == k) {
            if (sum == 0) {
                res.push_back(path);
            }
            return;
        }
        for (int i = begin; i <= 9; i++) {
            if (path.size() + (9 - i + 1) < k) { // 纵向剪枝
                break;
            }
            if (sum - i < 0) { // 横向剪枝
                break;
            }
            path.push_back(i);
            back(sum - i, k, i + 1);
            path.pop_back();
        }
        return;
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        res.clear();
        path.clear();
        back(n, k, 1);
        return res;
    }
};

其中,纵向上的剪枝是我自己没有考虑到的。


        第2题(17. 电话号码的字母组合)需要首先把每个数字按键和对应的多个字母记录整理下来,并同之前一样设置2个全局变量path和res。之后按照回溯的模板,递归函数参数设置为字母组合digits和当前下标ind。这里要注意,为了避免多次递归时函数参数的值传递导致效率下降,digits的参数类型设置为引用类型。这道题与第一题还有个不同在于每次循环的长度都是3或4,无论怎么选树的深度都会是字母组合的长度,这是因为是在多个结合中进行组合,不像前面的回溯题都是在同一集合中选取元素进行组合。

class Solution {
public:
    const string letters[10] = {
        "", 
        "", 
        "abc", 
        "def", 
        "ghi",
        "jkl", 
        "mno",
        "pqrs",
        "tuv",
        "wxyz"
    };
    vector<string> res;
    string path;
    void back(string& digits, int ind) {
        if (ind == digits.size()) {
            res.push_back(path);
            return;
        }
        int num = digits[ind] - '0';
        if (num < 2 || num > 9) {
            back(digits, ind + 1);
            return;
        }
        for (int i = 0; i < letters[num].size(); ++i) {
            path.push_back(letters[num][i]);
            back(digits, ind + 1);
            path.pop_back();
        }
        return;
    }
    vector<string> letterCombinations(string digits) {
        res.clear();
        if (digits.size() == 0) {
            return res;
        }
        path.clear();
        back(digits, 0);
        return res;
    }
};

这道题的题目描述中有说明digits长度有可能为0,所以需要为这种特殊情况做特殊处理。最后,其中第23~26行不需要也是可以AC的,但可能出现非2~9的符号(如*、1等),遇到这种情况应该跳过当前字符,跳过的方法就是不做任何处理,直接继续递归下一下标的符号。

        二刷:忘记处理输入digits为空的情况。忘记string.erase()的用法(第一个参数是开始删除的下标,第二个参数是要删除的长度,如s.erase(s.size() - 1, 1)表示删除s的最后一位)。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值