【leetcode】93. 复原 IP 地址（Java）

题目描述

题目链接93. 复原 IP 地址

题解

回溯法

• 我们的目的就是在这一个字符串中插入三个点，然后使每一段都在[0, 255]之间。
• 回溯终止条件：当插入的点为3个，并且最后一段在[0, 255]之间。
• 处理逻辑，当这一段有效的时候，在后面插入一个点，递归处理后面的字符串。详见代码注释

class Solution {
    List<String> res = new ArrayList<>();
    StringBuffer path = new StringBuffer();

    public List<String> restoreIpAddresses(String s) {
    	//特殊情况，直接返回
        if (s.length() < 4 || s.length() > 12) return res;
        //开始回溯
        dfs(s, 0, s.length() - 1, 0);
        return res;
    }

	//start标记开始的位置，end标记为结束的位置，count为小数点计数
    public void dfs(String s, int start, int end, int count){
    	//如果小数点为3个，并且最后一段有效的话，保存结果
        if (count == 3 && isValid(s.substring(start, end + 1))){
            res.add(path.append(s.substring(start, end + 1)).toString());
            return;
        }
        //从start开始，遍历右边界
        for (int i = start + 1; i <= end; i++){
        	//当[start, i]合法的时候，在i后面插入一个点，递归处理后面的字符串
            if (isValid(s.substring(start, i))){
            	//path添加[start, i]的字符串
                path.append(s.substring(start, i));
                //加入点，计数加一
                path.append('.');
                count++;
                //递归遍历后面的字符串
                dfs(s, i, end, count);
                //回溯记得归位
                count--;
                path.delete(start + count, path.length());
            }
        }
    }

	//判断字符串是否在[0,255]之间
    public boolean isValid(String s){
        if (s.length() == 0) return false;
        char[] cs = s.toCharArray();
        if (cs[0] == '0' && cs.length != 1) return false;
        int res = 0;
        for (int i = 0; i < cs.length; i++){
            res = res * 10 + cs[i] - '0';
        }
        if (res >= 0 && res <= 255){
            return true;
        }
        return false;
    }
}