【leetcode热题100】解码方法

• 难度： 中等
• 通过率： 21.5%
• 题目链接：力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

题目描述

一条包含字母 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 的解码方式，可以是在 s[0:-1] 的基础上，将 s[-1] 解码，这样 s 的解码结果数就等于 s[0:-1] 的解码结果数。

也可以在 s[0:-2] 的基础上，将 s[-2] 和 s[-1] 合在一起解码，如果能够解码成功，其解码结果数就等于对 s[0:-2] 解码的结果数。

class Solution:
    def numDecodings(self, s):
        """
        :type s: str
        :rtype: int
        """
        if len(s) == 0 or s[0] == '0':
            return 0
        
        if len(s) == 1:
            return 1
        
        a, b = 1, 1
        for i in range(2, len(s)):
            ans = 0
            if s[i] != '0':
                ans += b
            if s[i-1] == '1' or (s[i-1] == '2' and s[i] <= '6'):
                ans += a

            a = b
            b = ans

        return b