LeetCode 93 Restore IP Addresses

LeetCode 93 Restore IP Addresses

Given a string containing only digits, restore it by returning all possible valid IP address combinations.
For example:Given "25525511135",
return ["255.255.11.135", "255.255.111.35"]. (Order does not matter)

看到题目,和combination或是subset非常像,都是数组组合问题,所以直接想到了backtracing!!!然而问题来了,回溯的时候到底终止条件是什么?循环条件又是什么!!!

真的被自己蠢哭了,写了两遍都没有写对,发现是自己逻辑错误。。。

终止条件不难,可以在回溯的时候传输一个segm变量代表当前ip含有多少个segment了,当ip已经有4个segment时,即可添加到结果list中。

那在每次回溯时,到底到循环判断多少个元素呢?!

对于permutaion和subset和combination来说,一般都是要循环遍历整个数组(或字符串)。但对于combination K,即从n个数中选出k个数不同的combination的题目,回溯终止条件变成了index==k。

前面说了句废话。。。进入正题,那到底需要循环判断多少元素呢。

对于一个ip,它的segment最多只能有3位(0-255),因此在添加新的segment时,只需要考虑当前index到index+2这3位数字即可!!!

这里还有一个trick是,到底数组(或string)以及ip应该怎么传递!
可以考虑每次传入string的子串(去除当前已经考虑过的字符)。

代码:

public class Solution {
    public List restoreIpAddresses(String s) {
        List combs = new ArrayList<>();
        if (s.length() < 4 || s.length() > 12) return combs;
        dfs(s, combs, "", 0, 0);
        return combs;
    }
    
    public void dfs(String s, List combs, String addr, int segm, int index) {
        if (segm >= 3) {
            String seg = s.substring(index);
            if (isValidSeg(seg)) {
                String ip = addr + seg;
                combs.add(ip);
            }
            return;
        }
        
        for (int i = 1; i < 4 && index+i < s.length(); i++) {
            String seg = s.substring(index, index+i);
            if (isValidSeg(seg)) {
                String ip = addr + seg + ".";
                dfs(s, combs, ip, segm+1, index+i);
            }
        }
            
    }
    
    public boolean isValidSeg(String str) {
        if (str.charAt(0) == '0' && str.length() > 1) return false;
        else if (Integer.valueOf(str) > 255) return false;
        else return true;
    }
    
    public static void main(String[] args) {
        Solution solver = new Solution();
        String s1 = "25525511135";
        String s2 = "0000";
        for (String str : solver.restoreIpAddresses(s1)) {
            System.out.println(str);
        }
        for (String str : solver.restoreIpAddresses(s2)) {
            System.out.println(str);
        }
    }
    
    
}

参考
https://discuss.leetcode.com/topic/3919/my-code-in-java
https://discuss.leetcode.com/topic/53012/a-concise-and-easy-java-solution

你可能感兴趣的:(LeetCode 93 Restore IP Addresses)