93. 复原IP地址/C++

93. 复原IP地址/C++_第1张图片
我们知道,ip地址是由“.”隔开的4段0-255的数,因此每一段都是1-3位数。另外每一段不能出现诸如“012”这样的数。

class Solution {
	vector<string> vs;
	string str;
    
    //判断s是否是符合要求的一段
	bool isvalid(string s){
		//不能为空,也不能以0开头
		if (s.empty() || s.size() > 1 && s[0] == '0') 
            return false;
		int num = stoi(s);
		//范围在0~255
		return num >= 0 && num <= 255;
	}
	
	//s表示还剩的字符串,index表示正在处理的段数(从0开始)
	void dfs(string s, int index){
		//如果前4段都处理完了,s不剩字符了,就表示str符合要求,否则回退
		if (index == 4){
			if (s.empty()){
				vs.push_back(str);
			}
			return;
		}
		else{
			//如果处理的不是第1段,就要先加一个“.”
			if (index > 0)
				str += '.';
			//尝试选取之后的1~3位
			for (int i = 1; i <= 3 && i<=s.size(); ++i){
					string s1 = s.substr(0, i);
					if (isvalid(s1)) {
						str += s1;
						dfs(s.substr(i), index + 1);
						//回溯
						str.erase(str.size() - i);
					}
			}
			//回溯
			if (index > 0)
				str.erase(str.end() - 1);
		}
	}
	
public:
	vector<string> restoreIpAddresses(string s) {
		dfs(s, 0);
		return vs;
	}
};

你可能感兴趣的:(回溯法,LeetCode/C++)