代码随想录算法训练营第二十八天【回溯算法part04】 | 93.复原IP地址、78.子集、90.子集II

93.复原IP地址

题目链接：

力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

求解思路：

回溯三部曲

1. 递归参数：字符串s，记录起始位置的startIndex，添加的逗点数量pointNum
2. 递归终止条件：逗点数量pointNum为3（说明分成了四段）
3. 单层搜索的逻辑：截取子串，并判断子串是否合法（通过另写一个函数实现），如果合法就在子串后面加上逗点，如果不合法就break结束本层循环（注意不是break）；注意递归的下一层要从i+2开始（因为添加了逗点），同时pointNum要+1

判断子串是否合法的函数

• 段位以0为开头的数字不合法
• 段位里有非正整数字符不合法
• 段位如果大于255了不合法

图解

代码：

class Solution {
private:
    bool isVaild(const string& s, int start, int end){
        if (start > end)
            return false;
        // 0开头的数字不合法
        if (s[start] == '0' && start != end)
            return false;
        // num用来计算数字大小
        int num = 0;
        for (int i = start; i <= end; i++){
            // 非数字的字符不合法
            if (s[i] < '0' || s[i] > '9')
                return false;
            num = num * 10 + (s[i] - '0');
            // 数字大于255不合法
            if (num > 255)
                return false; 
            }
            return true;
        }
        vector<string> result;
        void backtracking(string& s, int startIndex, int pointNum){
            // 当逗点的数量为3，分割结束
            if (pointNum == 3){
                // 判断第四段字符串是否合法
                if (isVaild(s, startIndex, s.size()-1))
                    result.push_back(s);
                return;
            }
            for (int i = startIndex; i < s.size(); i++){
                // 判断[startIndex, i]这个子串是否合法
                if (isVaild(s, startIndex, i)){
                    // 在i的后面插入一个逗点
                    s.insert(s.begin() + i + 1, '.');
                    pointNum++;
                    // 下一个子串的起始位置为i+2（因为插入了一个逗点）
                    backtracking(s, i + 2, pointNum);
                    pointNum--;
                    s.erase(s.begin() + i + 1);
                }
                else
                    break;
            }
        }
public:
    vector<string> restoreIpAddresses(string s) {
        backtracking(s, 0, 0);
        return result;
    }
};

78.子集

题目链接：

力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

求解思路：

回溯三部曲

1. 递归函数参数：数组nums，起始编号startIndex
2. 终止条件：start大于数组的长度，此时已经没有元素可以取了
3. 单层搜索逻辑：遍历整棵树，收集树形结构中树的所有节点的结果

图解

代码：

class Solution {
private:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& nums, int startIndex){
        // 收集子集
        result.push_back(path);
        if (startIndex > nums.size()){
            return;
        }
        for (int i = startIndex; i < nums.size(); i++){
            path.push_back(nums[i]);
            backtracking(nums, i+1);
            path.pop_back();
        }
    }
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        result.clear();
        path.clear();
        backtracking(nums, 0);
        return result;
    }
};

90.子集II

题目链接：

力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

求解思路：

• 78.子集 中的收集所有节点结果
• 40.组合总和II 中的去重的逻辑

代码：

class Solution {
private:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& nums, int startIndex, vector<bool>& used){
        result.push_back(path);
        for (int i = startIndex; i < nums.size(); i++){
            if (i > 0 && nums[i] == nums[i-1] && used[i-1] == false)
                continue;
            path.push_back(nums[i]);
            used[i] = true;
            backtracking(nums, i+1, used);
            path.pop_back();
            used[i] = false;
        }
    }
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        result.clear();
        path.clear();
        vector<bool> used(nums.size(), false);
        sort(nums.begin(), nums.end());
        backtracking(nums, 0, used);
        return result;
    }
};