问题描述:
一条包含字母 A-Z 的消息通过以下方式进行了编码:
'A' -> 1
'B' -> 2
...
'Z' -> 26
给定一个只包含数字的非空字符串,请计算解码方法的总数。
示例 1:
输入: "12"
输出: 2
解释: 它可以解码为 "AB"(1 2)或者 "L"(12)。
示例 2:
输入: "226"
输出: 3
解释: 它可以解码为 "BZ" (2 26), "VF" (22 6), 或者 "BBF" (2 2 6) 。
解题思路:动态规划。确定状态转移方程s[i]=s[i+1]+s[i+2](新添加一位数字时,要判断单独编码和组合编码的总组合数),确定编码是否合法。
代码如下:
class Solution:
def numDecodings(self, s: str) -> int:
def help(i):
if i in dic:
return dic[i]
if i >= len(s):
return 1
res = 0
# 第一位合法的情况(0不合法)
if 1 <= int(s[i]) <= 9:
res += help(i + 1)
# 前两位合法的情况
if 10 <= int(s[i:i+2]) <= 26:
res += help(i + 2)
# 之所以需要dic, 是因为i + 1, i + 2的两次递归调用, 会产生重复计算
dic[i] = res
return res
dic = {}
return help(0) if s else 0
时间与空间复杂度: