天天看点

利用python 完成leetcode 91 解码方法

一条包含字母 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,dp[i]表示前i+1个字母有多少种解码方法,

前i个字母最后一个单词有两种选择,取倒数两个字母为一个单词或者一个字母为一个单词

如果s[i]可以解码dp[i]=dp[i]+dp[i-1]

如果s[i-1:i+1]可以解码dp[i]=dp[i]+dp[i-2]

代码

def numDecodings(self, s):
    l=[]
    for i in range(1,27):l.append(str(i))
    dp=[0]*(len(s))
    if(s[0]in l):dp[0]=1
    for i in range(1,len(s)):
        a=0
        b=0
        if(s[i]in l):a=dp[i-1]
        if  s[i-1:i+1] in l:
            if(i==1):b=1
            else: b=dp[i-2]
        dp[i]=a+b
    return dp[-1]