【Lintcode】426. Restore IP Addresses

题目地址：

https://www.lintcode.com/problem/restore-ip-addresses/description

给定一个由数字组成的字符串，返回其加上小数点组成的所有合法的IP地址。合法的IP地址由四个$0\sim 255$的数字组成。

思路是DFS。尝试切出一个子串，如果符合条件就加到StringBuilder末尾，然后进入下一层递归。注意几个可以剪枝的情况：只需枚举切出长度为$3$或更短的数字即可，此外如果发现剩余的长度大于$3$乘以剩余要切出数字个数，那也不用切了，因为继续切下去会切不到末尾。代码如下：

import java.util.ArrayList;
import java.util.List;

public class Solution {
    /**
     * @param s: the IP string
     * @return: All possible valid IP addresses
     */
    public List<String> restoreIpAddresses(String s) {
        List<String> res = new ArrayList<>();
    
        dfs(s, 0, new StringBuilder(), 0, res);
    
        return res;
    }
    
    private void dfs(String s, int pos, StringBuilder sb, int cut, List<String> res) {
        if (pos == s.length()) {
        	// 如果到达了末尾且切出了4个数，则加入最终结果。但要把结尾的小数点去掉
            if (cut == 4) {
                res.add(sb.substring(0, sb.length() - 1));
            }
            return;
        }
        
        for (int i = Math.min(s.length(), pos + 3); i >= pos + 1; i--) {
            if (s.length() - i > (4 - cut) * 3) {
                return;
            }
            
            String cur = s.substring(pos, i);
            if ((!cur.startsWith("0") || cur.equals("0")) && Integer.parseInt(cur) <= 255 && cut < 4) {
                sb.append(cur).append('.');
                dfs(s, i, sb, cut + 1, res);
                sb.setLength(sb.length() - cur.length() - 1);
            }
        }
    }
}

时间复杂度$O(n{2}^{\frac{n}{2}})$，空间复杂度$O(n)$。