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

93.复原IP地址

题目链接：93.复原IP地址

文档讲解：代码随想录

视频讲解：回溯算法如何分割字符串并判断是合法IP？| LeetCode：93.复原IP地址

思路

与131.分割回文串类似，寻找间隔点插入.。

首先写一个判断数值是否大于255的函数，函数中按照逐位乘10相加的形式，当数值大于255时立刻返回false，若直接使用字符串转数值，可能会溢出。

每次判断数值合法时，在间隔点插入.，如果已插入’.'的数量为3时，判断一下剩余的数值是否合法，若合法则将该分割法插入结果数组，否则继续回溯。

代码

class Solution {
public:
    vector<string> restoreIpAddresses(string s) {
        backtracking(s, 0, 0);
        return result;
    }

private:
    vector<string> result;
    void backtracking(string &s, int startIndex, int count) {
        if (count == 3) {
            if (isIP(s, startIndex, s.size() - 1))
                result.push_back(s);
            return;
        }

        for (int i = startIndex; i < s.size(); i++) {
            if (isIP(s, startIndex, i)) {
                s.insert(s.begin() + i + 1, '.');
                count++;
                backtracking(s, i + 2, count);
                count--;
                s.erase(s.begin() + i + 1);
            }
            else
                continue;
        }
    }

    // 判断字符串s在左闭右闭区间[start, end]所组成的数字是否合法
    bool isIP(const string &s, int start, int end) {
        if (start > end)
            return false;
        if (s[start] == '0' && start != end)
            return false;
        int num = 0;
        for (int i = start; i <= end; i++) {
            if (s[i] > '9' || s[i] < '0')
                return false;
            num = num * 10 + s[i] - '0';
            if (num > 255)
                return false;
        }
        return true;
    }
};

78.子集

题目链接：78.子集

文档讲解：代码随想录

视频讲解：回溯算法解决子集问题，树上节点都是目标集和！ | LeetCode：78.子集

思路

抽象为树形结构后，收集树的所有结点，注意不能重复选取。

代码

初始思路代码

class Solution {
public:
    vector<vector<int>> subsets(vector<int> &nums) {
        backtracking(nums, 0);
        return result;
    }

private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(const vector<int> &nums, int startIndex) {
        result.push_back(path);

        for (int i = startIndex; i < nums.size(); i++) {
            path.push_back(nums[i]);
            backtracking(nums, i + 1);
            path.pop_back();
        }
    }
};

90.子集II

题目链接：90.子集II

文档讲解：代码随想录

视频讲解：回溯算法解决子集问题，如何去重？| LeetCode：90.子集II

思路

与40.组合总和II数层去重类似，首先对数组进行排序，然后定义一个used变量存储数组每个元素是否使用，如果当前选取的元素与前一个元素相同，且前一个元素未被使用时，则重复。

代码

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int> &nums) {
        sort(nums.begin(), nums.end());
        used.resize(nums.size());
        backtracking(nums, 0);
        return result;
    }

private:
    vector<vector<int>> result;
    vector<int> path;
    vector<bool> used;
    void backtracking(const vector<int> &nums, int startIndex) {
        result.push_back(path);

        for (int i = startIndex; i < nums.size(); i++) {
            if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == 0)
                continue;
            path.push_back(nums[i]);
            used[i] = 1;
            backtracking(nums, i + 1);
            used[i] = 0;
            path.pop_back();
        }
    }
};