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<String> restoreIpAddresses(String s) {
List<String> 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<String> 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