LeetCode 91. 解码方法(动态规划)
程序员文章站
2022-07-05 18:17:24
...
Description
一条包含字母 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) 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/decode-ways
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
Solution
这个题目的转移方程比较特殊,当前状态是由前面状态的相加,而不是前面状态+1的最大值。谨慎考虑。
dp[i]:以s[i]结尾的解码数量
class Solution:
def numDecodings(self, s: str) -> int:
s = ' ' + s
dp = [0 for _ in range(len(s))]
dp[0] = 1 # 空串只有一种解析方法
dp[1] = 1 if s[1]!='0' else 0
for i in range(2, len(s)):
if s[i] != '0':
dp[i] += dp[i-1]
if '26' >= s[i-1]+s[i] >= '10':
dp[i] += dp[i-2]
return dp[-1]