LeetCode 91 解码方法

一条包含字母 A-Z 的消息通过以下映射进行了 编码 :

'A' -> "1"
'B' -> "2"
...
'Z' -> "26"

要 解码 已编码的消息,所有数字必须基于上述映射的方法,反向映射回字母(可能有多种方法)。例如,”11106″ 可以映射为:

"AAJF" ,将消息分组为 (1 1 10 6)
"KJF" ,将消息分组为 (11 10 6)

注意,消息不能分组为 (1 11 06) ,因为 “06” 不能映射为 “F” ,这是由于 “6” 和 “06” 在映射中并不等价。

给你一个只含数字的 非空 字符串 s ,请计算并返回 解码 方法的 总数 。

题目数据保证答案肯定是一个 32 位 的整数。

示例 1:

输入:s = "12"
输出:2
解释:它可以解码为 "AB"(1 2)或者 "L"(12)。

示例 2:

输入:s = "226"
输出:3
解释:它可以解码为 "BZ" (2 26), "VF" (22 6), 或者 "BBF" (2 2 6) 。

示例 3:

输入:s = "06"
输出:0
解释:"06" 无法映射到 "F" ,因为存在前导零("6" 和 "06" 并不等价)。

提示:

1 <= s.length <= 100
s 只包含数字,并且可能包含前导零。

解题:

func numDecodings(s string) int {
	r, l := 0, len(s)
	var dfs func(i int)
	dfs = func(i int) {
		if i == l {
			r++
			return
		}
		if s[i] == '0' {
			if i == 0 {
				return
			}
			if i > 0 && s[i-1] == '0' {
				return
			}
			dfs(i + 1)
			return
		}
		if i+1 < l {
			if s[i+1] == '0' {
				dfs(i + 2)
			} else if s[i] > '2' || (s[i] > '1' && s[i+1] > '6') || (i+2 < l && s[i+2] == '0') {
				dfs(i + 1)
			} else {
				dfs(i + 1)
				dfs(i + 2)
			}
		} else {
			dfs(i + 1)
		}
	}
	dfs(0)
	return r
}

超时了,看了答案才知道方向错了。。

官方解答:

1.动态规划

func numDecodings(s string) int {
	n := len(s)
	f := make([]int, n+1)
	f[0] = 1
	for i := 1; i <= n; i++ {
		if s[i-1] != '0' {
			f[i] += f[i-1]
		}
		if i > 1 && s[i-2] != '0' && ((s[i-2]-'0')*10+(s[i-1]-'0') <= 26) {
			f[i] += f[i-2]
		}
	}
	return f[n]
}

注意到在状态转移方程中,f(i)的值仅与f(i-1)和f(i-2)有关,因此我们可以使用三个变量进行状态转移,省区数组的空间。

func numDecodings(s string) int {
	n := len(s)
	//a = f[i-2], b = f[i-1], c = f[i]
	a, b, c := 0, 1, 0
	for i := 1; i <= n; i++ {
		c = 0
		if s[i-1] != '0' {
			c += b
		}
		if i > 1 && s[i-2] != '0' && ((s[i-2]-'0')*10+(s[i-1]-'0') <= 26) {
			c += a
		}
		a, b = b, c
	}
	return c
}

最后验证了一下,超时的那个在本地和官方解法的答案是一致的

发表回复

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