(回溯法)ip地址的合理性

Kobe10發表於2017-02-03
  • 題目:
    給定一個只包含數字的字串,通過返回所有可能有效的IP地址組合來恢復它。
    
    例如:
    給定“25525511135”,
    
    return [“255.255.11.135”,“255.255.111.35”]。 (順序無所謂)

     

  • 程式碼:
    class Solution {
    public:
        vector<string> restoreIpAddresses(string s) {
            vector<string> res;
            restore(s, 4, "", res);
            return res;
        }
        void restore(string s, int k, string out, vector<string> &res) {
            if (k == 0) {
                if (s.empty()) res.push_back(out);
            }
            else {
                for (int i = 1; i <= 3; ++i) {
                    if (s.size() >= i && isValid(s.substr(0, i))) {
                        if (k == 1) restore(s.substr(i), k - 1, out + s.substr(0, i), res);
                        else restore(s.substr(i), k - 1, out + s.substr(0, i) + ".", res);
                    }
                }
            }
        }
        bool isValid(string s) {
            if (s.empty() || s.size() > 3 || (s.size() > 1 && s[0] == '0')) return false;
            int res = atoi(s.c_str());
            return res <= 255 && res >= 0;
        }
    };

     

相關文章