leetcode7-复原IP地址

题目:

 首先胖博主需要说明的是,IP地址字符串长度小于12,每三位不得大于255.。。。

代码参考https://www.cnblogs.com/ariel-dreamland/p/9159611.html

       如果k = 0,则表示三个点已经加入完成,四段已经形成,若这时字符串刚好为空,则将当前分好的结果保存。若k != 0, 则对于每一段,我们分别用一位,两位,三位来尝试,分别判断其合不合法,如果合法,则调用递归继续分剩下的字符串,最终和求出所有合法组合。

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;
   }
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);
			 }
	    }

	}
}

vector<string> restoreIpAddresses(string s) 
{
	vector<string>res;
	//先判断大小
	if (s.size()>12)
	{
		return res;
	}
	restore(s,4,"",res);
	return res;
}

 或者再简化一点

省掉isValid函数,直接在调用递归之前用if语句来滤掉不符合题意的情况,这里面用了k != std::to_string(val).size(),其实并不难理解,比如当k=3时,说明应该是个三位数,但如果字符是"010",那么转为整型val=10,再转回字符串就是"10",此时的长度和k值不同了,这样就可以找出不合要求的情况了。

class Solution {
public:
    vector<string> restoreIpAddresses(string s) {
        vector<string> res;
        helper(s, 0, "", res);
        return res;
    }
    void helper(string s, int n, string out, vector<string>& res) {
        if (n == 4) {
            if (s.empty()) res.push_back(out);
        } else {
            for (int k = 1; k < 4; ++k) {
                if (s.size() < k) break;
                int val = atoi(s.substr(0, k).c_str());
                if (val > 255 || k != std::to_string(val).size()) continue;
                helper(s.substr(k), n + 1, out + s.substr(0, k) + (n == 3 ? "" : "."), res);
            }
        }
    }
};

猜你喜欢

转载自blog.csdn.net/hi_baymax/article/details/82586417