一条包含字母 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
}
最后验证了一下,超时的那个在本地和官方解法的答案是一致的