Leetcode 93. 复原 IP 地址

93. 复原 IP 地址

有效 IP 地址 正好由四个整数（每个整数位于 0 到 255 之间组成，且不能含有前导 0），整数之间用 ‘.’ 分隔。

例如：“0.1.2.201” 和 “192.168.1.1” 是 有效 IP 地址，但是 “0.011.255.245”、“192.168.1.312” 和 “192.168@1.1” 是 无效 IP 地址。
给定一个只包含数字的字符串 s ，用以表示一个 IP 地址，返回所有可能的有效 IP 地址，这些地址可以通过在 s 中插入 ‘.’ 来形成。你 不能 重新排序或删除 s 中的任何数字。你可以按 任何 顺序返回答案。

示例 1：

输入：s = “25525511135”
输出：[“255.255.11.135”,“255.255.111.35”]

示例 2：

输入：s = “0000”
输出：[“0.0.0.0”]

示例 3：

输入：s = “101023”
输出：[“1.0.10.23”,“1.0.102.3”,“10.1.0.23”,“10.10.2.3”,"101.0.2.3

代码

class Solution {
public:
    vector<string> res;
    // 不满足条件
    // 1.前导0
    // 2.非数字
    // 3.大于255
    bool isValid(string& s, int st, int ed) {
        if (st > ed) return false;
 
        if (s[st] == '0' && st != ed)  // 前导 0 
            return false; 
        
        int sum = 0;
        for (int i = st; i <= ed; i++) {
            sum = sum * 10 + s[i] - '0';
            if (sum > 255) return false;
        }

        return true;
    }
    void backtracking(string& s, int index, int cnt) { // index 搜索的起始位置，cnt：加的.的总数
        if (cnt > 3) return;
        if (cnt == 3) { // 三个点就可以分成四段了
            if (isValid(s, index, s.size() - 1)) { // 只需判断最后一段符不符合就好，因为前面for循环也会判断一下，只有满足的前三段才能进行递归 
                res.push_back(s);
            }
            return;
        }

        for (int i = index; i < s.size(); i++) {
            // 直接修改字符串s
            if (isValid(s, index, i)) { // 满足条件
                s.insert(s.begin() + i + 1, '.');
                cnt++;
            }else {
                break; // 不满足条件直接结束循环
            }

            backtracking(s, i + 2, cnt); // i + 2 因为i 后面插入了一个.
            cnt--; // 回溯
            s.erase(s.begin() + i + 1); // 回溯，删除插入的点

        }
    }
    vector<string> restoreIpAddresses(string s) {
        res.clear();
        if (s.size() < 4) return res;
        backtracking(s, 0, 0);
        return res;
    }
};