lintcode-【中等】恢复IP地址

题目

给一个由数字组成的字符串。求出其可能恢复为的所有IP地址。链接。

样例

给出字符串 "25525511135"，所有可能的IP地址为：

[
  "255.255.11.135",
  "255.255.111.35"
]

答案

直接暴力遍历就行了，只不过需要注意的是0，以及数字是不能有前缀0。

代码

class Solution {
private:
    int strToInt(const string &str)
    {
        int ans = 0;
        int len = str.size();
        
        if(len > 1 && str[0] == '0')
        {
            return 256;
        }
        
        for(int i = 0; i < len; ++ i)
        {
            ans = (ans * 10 + str[i] - '0');
        }
        
        return ans;
    }
    
public:
    /**
     * @param s the IP string
     * @return All possible valid IP addresses
     */
    vector<string> restoreIpAddresses(string& s) {
        // Write your code here
        int len = s.size();
        vector<string> ans;
        string result;
        
        string first;
        string second;
        string third;
        string fourth;
        int iFirst,iSecond,iThird,iFourth;
        for(int i = 0;i < len - 3; ++ i)
        {
            first.push_back(s[i]);
            iFirst = strToInt(first);
            if(iFirst > 255)
            {
                break;
            }
            
            second.clear();
            for(int j = i + 1; j < len - 2; ++ j)
            {
                second.push_back(s[j]);
                iSecond = strToInt(second);
                if(iSecond > 255)
                {
                    break;
                }
                
                third.clear();
                for(int k = j + 1;k < len - 1; ++ k)
                {
                    third.push_back(s[k]);
                    iThird = strToInt(third);
                    if(iThird > 255)
                    {
                        break;
                    }
                    
                    fourth.clear();
                    for(int l = k + 1;l < len; ++ l)
                    {
                        fourth.push_back(s[l]);
                    }
                    iFourth = strToInt(fourth);
                    if(iFourth <= 255)
                    {
                        result = first + "." + second + "." + third + "." + fourth;
                        ans.push_back(result);
                    }
                        
                    
                }
            }
        }
        
        return ans;
    }
};

我自己写的字符串转数字的方法，尽管知道可以使用sstream转换或者说使用atoi，而且以前都是使用sstream，但是还是想自己动动手玩玩。