mirror of
https://github.com/halfrost/LeetCode-Go.git
synced 2025-07-05 00:25:22 +08:00
50 lines
906 B
Go
50 lines
906 B
Go
package leetcode
|
|
|
|
import (
|
|
"strconv"
|
|
)
|
|
|
|
func restoreIPAddresses(s string) []string {
|
|
if s == "" {
|
|
return []string{}
|
|
}
|
|
res, ip := []string{}, []int{}
|
|
dfs(s, 0, ip, &res)
|
|
return res
|
|
}
|
|
|
|
func dfs(s string, index int, ip []int, res *[]string) {
|
|
if index == len(s) {
|
|
if len(ip) == 4 {
|
|
*res = append(*res, getString(ip))
|
|
}
|
|
return
|
|
}
|
|
if index == 0 {
|
|
num, _ := strconv.Atoi(string(s[0]))
|
|
ip = append(ip, num)
|
|
dfs(s, index+1, ip, res)
|
|
} else {
|
|
num, _ := strconv.Atoi(string(s[index]))
|
|
next := ip[len(ip)-1]*10 + num
|
|
if next <= 255 && ip[len(ip)-1] != 0 {
|
|
ip[len(ip)-1] = next
|
|
dfs(s, index+1, ip, res)
|
|
ip[len(ip)-1] /= 10
|
|
}
|
|
if len(ip) < 4 {
|
|
ip = append(ip, num)
|
|
dfs(s, index+1, ip, res)
|
|
ip = ip[:len(ip)-1]
|
|
}
|
|
}
|
|
}
|
|
|
|
func getString(ip []int) string {
|
|
res := strconv.Itoa(ip[0])
|
|
for i := 1; i < len(ip); i++ {
|
|
res += "." + strconv.Itoa(ip[i])
|
|
}
|
|
return res
|
|
}
|