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