LeetCode-639、解码方法 2-困难

最新推荐文章于 2023-02-24 17:42:38 发布

clover_my

最新推荐文章于 2023-02-24 17:42:38 发布

阅读量253

点赞数

分类专栏： LeetCode题库

本文链接：https://blog.csdn.net/clover_my/article/details/103602001

版权

LeetCode题库专栏收录该内容

218 篇文章 6 订阅

订阅专栏

LeetCode-639、解码方法 2-困难

一条包含字母 A-Z 的消息通过以下的方式进行了编码：

'A' -> 1
'B' -> 2
...
'Z' -> 26

除了上述的条件以外，现在加密字符串可以包含字符 ' * ' 了，字符 ' * ' 可以被当做1到9当中的任意一个数字。

给定一条包含数字和字符 ' * ' 的加密信息，请确定解码方法的总数。

同时，由于结果值可能会相当的大，所以你应当对 10^9 + 7 取模。（翻译者标注：此处取模主要是为了防止溢出）

示例 1 :

输入: "*"
输出: 9
解释: 加密的信息可以被解密为: "A", "B", "C", "D", "E", "F", "G", "H", "I".

示例 2 :

输入: "1*"
输出: 9 + 9 = 18（翻译者标注：这里1*可以分解为1,* 或者当做1*来处理，所以结果是9+9=18）

说明 :

输入的字符串长度范围是 [1, 105]。
输入的字符串只会包含字符 '*' 和 数字'0' - '9'。

代码：

class Solution:
    def numDecodings(self, s: str) -> int:
        if not s or s[0] == "0":
            return 0
        M = 10 ** 9 + 7
        pre = 1
        cur = 9 if s[0] == "*" else 1
        for i in range(1, len(s)):
            tmp = cur
            if s[i] == "0":
                if s[i-1] == "1" or s[i-1] == "2": cur = pre
                elif s[i-1] == "*": cur = pre * 2 % M
                else: return 0
            elif s[i] == "*":
                if s[i-1] == "1": cur = (pre * 9 + cur * 9) % M
                elif s[i-1] == "2": cur = (pre * 6 + cur * 9) % M
                elif s[i-1] == "*": cur = (pre * 15 + cur * 9) % M
                else: cur = cur * 9 % M
            else:
                if s[i-1] == "1" or (s[i-1] == "2" and int(s[i]) <= 6): cur = (cur + pre) % M
                elif s[i-1] == "*":
                    if int(s[i]) <= 6: cur = (cur + pre * 2) % M
                    elif int(s[i]) > 6: cur = (cur + pre) % M
            pre = tmp
        return cur