LeetCode 93. 復原IP地址（Restore IP Addresses）

題目描述

 

給定一個只包含數字的字符串，復原它並返回所有可能的 IP 地址格式。

示例:

輸入: "25525511135"
輸出: ["255.255.11.135", "255.255.111.35"]

 

解題思路

 

利用回溯法的思想，從字符串第一個字符開始，分別檢查從當前字符開始的第1、2、3位組成的數，若該數符合要求即小於256則加入到結果IP中，然后再從當前數的后一位置遞歸向后尋找。若遍歷到第四個數，則判斷最后幾位組成的數是否滿足要求，符合則加入到最后的集合中。注意當遇到當前數的起始位是0時，僅將0加入到字符串中就停止當前數的尋找，因為以‘0’開頭的任何字符串轉化成的數都會去掉前面的‘0’

 

代碼

 

class Solution {
public:
    vector<string> restoreIpAddresses(string s) {
        vector<string> res;
        findIp(s, 0, 0, "", res);
        return res;
    }
    void findIp(string s, int f, int idx, string ip, vector<string> &res){
        if(idx == 3){
            if(s.size() - 1 - f < 3){
                if(s[f] == '0' && f != s.size() - 1) return;
                int num = stoi(s.substr(f, s.size() - f)); 
                if(num < 256){
                    ip += to_string(num);
                    res.push_back(ip);
                }
            }
        }
        else{
            for(int i = 1; i <= 3; i++){
                if(f + i >= s.size()) break;
                int num = stoi(s.substr(f, i));
                if(num < 256)
                    findIp(s, f + i, idx + 1, ip + to_string(num) + ".", res);
                if(s[f] == '0' && i == 1) break;
            }
        }
    }
};