题目分析:
一条包含字母 A-Z 的消息通过以下方式进行了编码:‘A’ -> 1,B’ -> 2, … ,‘Z’ -> 26。给定一个只包含数字的非空字符串,请计算解码方法的总数。如:
输入: “12”
输出: 2
解释: 它可以解码为 “AB”(1 2)或者 “L”(12)。
解题思路:
把这个题可以分解为几种小问题进行解决,一般涉及字符串求几种情况,或者最优情况便可用动态规划。其实这个题只有四种情况:
- 以1或2开头后面有非0数字,那么结果就是去掉上一个字符的所有情况 + 去掉上两个字符的所有情况
- 以1或2开头后面是0,那么结果就是去掉上两个字符的所有情况
- 当前数字非0,那么结果就是去掉上一个字符的所有情况
- 连续出现0,那么不能解码,返回0
写成代码就是
if (s[i - 2: i ] > '10' and s[i - 2: i] <= '26') and s[i - 1] != '0':
dp.append(dp[i - 1] + dp[i - 2])
elif s[i - 2: i] == '10' or s[i - 2: i] == '20':
dp.append(dp[i - 2])
elif s[i - 1] != '0':
dp.append(dp[i - 1])
else:
return 0
测试代码:(Runtime: 40 ms, faster than 94.21% )
class Solution:
def numDecodings(self, s):
if s[0] == '0': return 0
dp=[1,1]
for i in range(2, len(s) + 1):
if (s[i - 2: i ] > '10' and s[i - 2: i] <= '26') and s[i - 1] != '0':
dp.append(dp[i - 1] + dp[i - 2])
elif s[i - 2: i] == '10' or s[i - 2: i] == '20':
dp.append(dp[i - 2])
elif s[i - 1] != '0':
dp.append(dp[i - 1])
else:
return 0
return dp[len(s)]
print(Solution().numDecodings("12")) #提交时请删除该行
附: 这个题目我一开始用的递归,也可以解出来,但是TLE,附上递归代码
class Solution:
def numDecodings(self, s: str) -> int:
self.n = 0
def dfs(s):
if s == '':
self.n += 1
elif (s[0] == '1') and 1 < len(s) or (s[0] == '2' and 1 < len(s) and s[1] <= '6'):
dfs(s[1:])
dfs(s[2:])
elif s[0] == '0':
return
else:
dfs(s[1:])
dfs(s)
return self.n
print(Solution().numDecodings("12")) #提交时请删除该行