题目:
一条包含字母 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]
代码:
class Solution(object):
def numDecodings(self, s):
"""
:type s: str
:rtype: int
"""
dic = list()
for i in range(1,27):
dic.append(str(i))
dp = [0]*len(s)
if s[0] in dic:
dp[0] = 1
else:
return 0
for i in range(1,len(s)):
a = 0
b = 0
if s[i] in dic:
a = dp[i-1]
if s[i-1:i+1] in dic:
if i == 1:
b = 1
else:
b = dp[i-2]
dp[i] = a+b
print(dp)
return dp[-1]