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

发布时间 2023-09-03 11:31:37作者: 银河小船儿
 

216.组合总和III 

     卡哥建议:如果把 组合问题理解了,本题就容易一些了。 

    题目链接/文章讲解:https://programmercarl.com/0216.%E7%BB%84%E5%90%88%E6%80%BB%E5%92%8CIII.html

    视频讲解:https://www.bilibili.com/video/BV1wg411873x

    做题思路:

   本题k相当于树的深度,9(因为整个集合就是9个数)就是树的宽度。

   例如 k = 2,n = 4的话,就是在集合[1,2,3,4,5,6,7,8,9]中求 k(个数) = 2, n(和) = 4的组合。

   选取过程如图:

216.组合总和III

    图中,可以看出,只有最后取到集合(1,3)和为4 符合条件。

   回溯三部曲看卡哥文章吧。

   剪枝优化:

   1,已选元素总和如果已经大于n(图中数值为4)了,那么往后遍历就没有意义了,直接剪掉。

   2,for循环的范围也可以剪枝,i <= 9 - (k - path.size()) + 1就可以了,和昨天的题一样

    本题代码:

 1 class Solution {
 2 private:
 3     vector<vector<int>> result; // 存放结果集
 4     vector<int> path; // 符合条件的结果
 5     void backtracking(int targetSum, int k, int sum, int startIndex) {
 6         if (sum > targetSum) { // 剪枝操作
 7             return; // 如果path.size() == k 但sum != targetSum 直接返回
 8         }
 9         if (path.size() == k) {
10             if (sum == targetSum) result.push_back(path);
11             return;
12         }
13         for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) { // 剪枝
14             sum += i; // 处理
15             path.push_back(i); // 处理
16             backtracking(targetSum, k, sum, i + 1); // 注意i+1调整startIndex
17             sum -= i; // 回溯
18             path.pop_back(); // 回溯
19         }
20     }
21 
22 public:
23     vector<vector<int>> combinationSum3(int k, int n) {
24         result.clear(); // 可以不加
25         path.clear();   // 可以不加
26         backtracking(n, k, 0, 1);
27         return result;
28     }
29 };

 

 17.电话号码的字母组合 

     卡哥建议:本题大家刚开始做会有点难度,先自己思考20min,没思路就直接看题解。 

    题目链接/文章讲解:https://programmercarl.com/0017.%E7%94%B5%E8%AF%9D%E5%8F%B7%E7%A0%81%E7%9A%84%E5%AD%97%E6%AF%8D%E7%BB%84%E5%90%88.html

    视频讲解:https://www.bilibili.com/video/BV1yV4y1V7Ug

    做题思路:

     电话按键上2~9数字上对应着3个或4个字母,题目求的是输入数字,输出字母的组合,可以使用map或者定义一个二维数组,例如:string letterMap[10][],来做映射,10行,4列的二维数组,代码见卡哥文章。

 

     图中可以看出遍历的深度,就是输入"23"的长度,而叶子节点就是我们要收集的结果,输出["ad", "ae", "af", "bd", "be", "bf", "cd", "ce", "cf"]。

   需要一个字符串s来收集叶子节点的结果,然后用一个字符串数组result保存起来,   
   代码中的 index 是从0开始的,是记录遍历第几个数字了,就是用来遍历digits的(题目中给出数字字符串“23”),就是下标,同时index也表示树的深度。 

   本题代码:

 1 class Solution {
 2 private:
 3     const string letterMap[10] = { //二维数组,10行,列并没有定义,但能看出来
 4         "", // 0
 5         "", // 1
 6         "abc", // 2
 7         "def", // 3
 8         "ghi", // 4
 9         "jkl", // 5
10         "mno", // 6
11         "pqrs", // 7
12         "tuv", // 8
13         "wxyz", // 9
14     };
15 public:
16     vector<string> result;//放叶子节点的结果
17     string s;
18     void backtracking(const string& digits, int index) {
19         if (index == digits.size()) {
20             result.push_back(s);
21             return;
22         }
23         int digit = digits[index] - '0';        // 将index指向的数字转为int,比如digits[0]="2",减完后,就是2了
24         string letters = letterMap[digit];      // 取数字对应的字符集,比如,letterMap[2]="abc",然后在二维数组里就能找到对应的字符串
25         for (int i = 0; i < letters.size(); i++) { //树的结构,看上图
26             s.push_back(letters[i]);            // 处理
27             backtracking(digits, index + 1);    // 递归,注意index+1,一下层要处理下一个数字了
28             s.pop_back();                       // 回溯
29         }
30     }
31     vector<string> letterCombinations(string digits) {
32         s.clear();
33         result.clear();
34         if (digits.size() == 0) {
35             return result;
36         }
37         backtracking(digits, 0);
38         return result;
39     }
40 };