题目:
一条包含字母 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) 。
思路:动态规划。
dp[i]表示s中以s[i]结尾的字符的解码数。更新规则如下:
1.若s[i] == "0"
,向前看一位:
- 如果前一位是1或者2:说明0可以和前一位结合,转移方程为:
若i-2>=0:dp[i] = dp[i-2]
,
否则:dp[i] = 1
; - 否则,
dp[i] = 0
。
2.否则,向前看一位:
- 如果前一位是1 或者 前一位是2并且当前位在1(含)和6(含)之间(即当前位和前一位可以结合成一个字母):
i-2>=0时:dp[i] = dp[i - 1] + dp[i - 2]
,dp[i - 1]
理解为从i
的前面断开,dp[i - 2]
理解为从i-1
的前面断开。
否则:dp[i] = dp[i - 1] + 1
- 否则:
dp[i] = dp[i - 1]
代码:
class Solution(object):
def numDecodings(self, s):
"""
:type s: str
:rtype: int
"""
l = len(s)
dp = [0] * l
dp[0] = 1 if int(s[0]) > 0 else 0
for i in range(1, l):
if s[i] == "0":
if s[i-1] == "1" or s[i-1] == "2":
if i-2>=0:
dp[i] = dp[i-2]
else:
dp[i] = 1
else:
dp[i] = 0
else:
if s[i-1] == "1" or (s[i-1] == "2" and s[i]>="1" and s[i]<="6"):
if i - 2 >= 0:
dp[i] = dp[i - 1] + dp[i - 2]
else:
dp[i] = dp[i - 1] + 1
else:
dp[i] = dp[i - 1]
return dp[l - 1]