LeetCode 93 复原 IP 地址

有效 IP 地址 正好由四个整数(每个整数位于 0 到 255 之间组成,且不能含有前导 0),整数之间用 ‘.’ 分隔。

例如:”0.1.2.201″ 和 “192.168.1.1” 是 有效 IP 地址,但是 “0.011.255.245”、”192.168.1.312″ 和 “192.168@1.1” 是 无效 IP 地址。
给定一个只包含数字的字符串 s ,用以表示一个 IP 地址,返回所有可能的有效 IP 地址,这些地址可以通过在 s 中插入 ‘.’ 来形成。你 不能 重新排序或删除 s 中的任何数字。你可以按 任何 顺序返回答案。

示例 1:

输入:s = "25525511135"
输出:["255.255.11.135","255.255.111.35"]

示例 2:

输入:s = "0000"
输出:["0.0.0.0"]

示例 3:

输入:s = "101023"
输出:["1.0.10.23","1.0.102.3","10.1.0.23","10.10.2.3","101.0.2.3"]

提示:

  • 1 <= s.length <= 20
  • s 仅由数字组成

解题:

func restoreIpAddresses(s string) []string {
	r := make([]string, 0)
	l := len(s)
	if l > 12 {
		return r
	}
	a := make([]string, 0)
	var dfs func(i, j int)
	dfs = func(i, j int) {
		if i == l && len(a) == 4 {
			r = append(r, strings.Join(a, "."))
			return
		}
		if i >= l || len(a) >= 4 {
			return
		}
		temp := ""
		if s[i] != '0' {
			if i+3 <= l {
				temp = s[i : i+3]
				tempInt, _ := strconv.Atoi(temp)
				if tempInt >= 0 && tempInt <= 255 {
					a = a[:j-1]
					a = append(a, temp)
					dfs(i+3, j+1)
				}
			}
			if i+2 <= l {
				temp = s[i : i+2]
				a = a[:j-1]
				a = append(a, temp)
				dfs(i+2, j+1)
			}
		}
		temp = s[i : i+1]
		a = a[:j-1]
		a = append(a, temp)
		dfs(i+1, j+1)
		return
	}
	dfs(0, 1)
	return r
}

官方解答:

1.回溯

const SegCount = 4

var (
	ans      []string
	segments []int
)

func restoreIpAddresses(s string) []string {
	segments = make([]int, SegCount)
	ans = []string{}
	dfs(s, 0, 0)
	return ans
}

func dfs(s string, segId, segStart int) {
	// 如果找到了 4 段 IP 地址并且遍历完了字符串,那么就是一种答案
	if segId == SegCount {
		if segStart == len(s) {
			ipAddr := ""
			for i := 0; i < SegCount; i++ {
				ipAddr += strconv.Itoa(segments[i])
				if i != SegCount-1 {
					ipAddr += "."
				}
			}
			ans = append(ans, ipAddr)
		}
		return
	}
	// 如果还没有找到 4 段 IP 地址就已经遍历完了字符串,那么提前回溯
	if segStart == len(s) {
		return
	}
	// 由于不能有前导零,如果当前数字为 0,那么这一段 IP 地址只能为 0
	if s[segStart] == '0' {
		segments[segId] = 0
		dfs(s, segId+1, segStart+1)
	}
	// 一般情况,枚举每一种可能性并递归
	addr := 0
	for segEnd := segStart; segEnd < len(s); segEnd++ {
		addr = addr*10 + int(s[segEnd]-'0')
		if addr > 0 && addr <= 0xFF {
			segments[segId] = addr
			dfs(s, segId+1, segEnd+1)
		} else {
			break
		}
	}
}

做了两个多小时,比官方速度快,值了

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注