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

LeetCode 91. 解码方法(动态规划)

程序员文章站 2022-07-15 18:06:55
...

题目描述

一条包含字母 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) 。

思路

详见链接

代码

class Solution:
	def numDecodings(self,s:str) -> int:
		n = len(s)
		if (not s or s[0] == "0"):
			return 0
		dp = [0]*(n+1)
		dp[0] = 1
		dp[1] = 1
		for i in range(1,n):
			if (s[i] == "0"):
				if(s[i-1] == "1" or s[i-1] == "2"):
					dp[i+1] = dp[i-1]
				else:
					return 0
			else:
				if(s[i-1] == "1" or s[i-1] == "2" and "1"<=s[i]<="6"):
					dp[i+1] = dp[i] + dp[i-1]
				else:
					dp[i+1] = dp[i]
		return dp[-1]
test = Solution()
test.numDecodings("12")

效果

LeetCode 91. 解码方法(动态规划)