欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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]