算法训练营Day25 第七章 回溯算法part2

第一题： 216.组合总和III 

        学习了昨天的那道组合（leetcode77）题目之后，可以使用相似的方式来解题。同样按照“回溯三部曲”的方式来进行。

        ①确定函数的参数与返回值：回溯函数写成形如void backtracking(int k, int n, int start)的格式，其中start表示这次搜索过程中开始的位置，它的作用是避免出现重复的组合（因为组合不讲究顺序）。

        ②确定终止条件：当sum大于n时，需要return；当path.size() == k时，如果又有sum == n，则将此时的path放入result中，然后再return，如果这时sum != n，那么直接return。

        ③确定单层搜索过程：单层搜索的过程与之前那道题类似，只不过每次将一个数字放入path的同时，还需要将这个数加给sum，而且在回溯的时候，不仅要pop尾部元素，也需要从sum中减去相应的数字。

        ④剪枝的优化：在第二步中，当sum > n的时候直接return就是剪枝的操作，这时path.size()小于k，但sum已经大于n了，说明这个结果之后不会符合条件了，直接剪枝就OK。还有一个剪枝的操作，与上一题是类似的，当剩余需要插入path的数字不足以满足k的需求时，那就直接return吧，因为这个结果中的个数最后无法达到k个了，因此剪枝就好了。

以下给出具体的代码：

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    int sum = 0;

    void backtracking(int k, int n, int start) {
        if(sum > n) return;
        if(path.size() == k) {
            if(sum == n) result.push_back(path);
            return;
        }
        for(int i = start; i <= 9 - (k - path.size()) + 1; ++i) {
            path.push_back(i);
            sum += i;
            backtracking(k, n, i + 1);
            path.pop_back();
            sum -= i;
        }
        return;
    }

    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(k, n, 1);
        return result;
    }
};

 第二题： 17.电话号码的字母组合

        这道题与前两道组合类的题目有所差别，这道题目中每个数字相当于一个独立的集合，是在多个集合中寻找不同的字母组合，而之前两道题相当于在同一个集合中找出不同的组合。因此，这道题不需要使用之前使用的start来避免出现重复的组合。

        解题思路：首先利用hashtable记录2-9这几个数字分别对应的字母字符串，并利用一个vector<string>来记录传入的数字（可能有多个）所对应的字母字符串。然后按照“回溯三部曲”来解题。

解题过程为：

①确定回溯函数的参数与返回值：写出的回溯函数形如：void backtracking(int n, vector<string>& vec, int index)。n代表传入数字的个数，vec代表这些数字对应的字符串，index用来表示当前遍历的字符串。

②确定终止条件：当n==path.size()时，此时就说明找到了一条结果，就需要把这条结果插入到结果集中，并返回。

③确定单层搜索过程：for循环可以从i=0开始了，这是与之前两道题目不同的地方，因为这次是多个集合，对于不同的集合我们每次都从第一个元素开始遍历才是正确的。将当前遍历的字符串中的第i个元素放入path中，接着递归地调用回溯函数，传入的index需要加1，之后还需要回溯操作，弹出最后一个元素。

几个需要注意的小细节：

①构造unordered_map时，内部也应该使用大括号{}，一开始写成了中括号就出错了！

②当digits.size() == 0时，应当直接返回result，不然返回的结果是一个[""]，而预期结果是[]。这是因为把一个空的string插入到结果中了，因而出现了问题。所以这里要直接返回哦！

整体代码如下所示：

class Solution {
public:
    vector<string> result;
    string path;

    unordered_map<char,string> mp = {
        {'2', "abc"},
        {'3', "def"},
        {'4', "ghi"},
        {'5', "jkl"},
        {'6', "mno"},
        {'7', "pqrs"},
        {'8', "tuv"},
        {'9', "wxyz"},
    };

    void backtracking(int n, vector<string>& vec, int index) {
        if(path.size() == n) {
            result.push_back(path);
            return;
        }
        for(int i = 0; i < vec[index].size(); ++i) {
            path.push_back(vec[index][i]);
            backtracking(n, vec, index + 1);
            path.pop_back();
        }
        return;
    }

    vector<string> letterCombinations(string digits) {
        int n = digits.size();
        if(n == 0) return result;
        vector<string> vec(n);
        for(int i = 0; i < n; ++i) {
            vec[i] = mp[digits[i]];
        }
        backtracking(n, vec, 0);
        return result;
    }
};

        Day25打卡！！！