一条包含字母 A-Z 的消息通过以下方式进行了编码:
'A' -> 1
'B' -> 2
...
'Z' -> 26
给定一个只包含数字的非空字符串,请计算解码方法的总数。
题目数据保证答案肯定是一个 32 位的整数。
示例 1:
输入:"12"
输出:2
解释:它可以解码为 "AB"(1 2)或者 "L"(12)。
示例 2:
输入:"226"
输出:3
解释:它可以解码为 "BZ" (2 26), "VF" (22 6), 或者 "BBF" (2 2 6) 。
示例 3:
输入:s = "0"
输出:0
示例 4:
输入:s = "1"
输出:1
示例 5:
输入:s = "2"
输出:1
提示:
1 <= s.length <= 100
s 只包含数字,并且可以包含前导零。
我们考虑用 dp[i]来切分子问题, 那么 dp[i]表示的意思是当前字符串的以索引 i 结尾的子问题。这样的话,我们最后只需要取 dp[s.length] 就可以解决问题了。
/**
* @param {string} s
* @return {number}
*/
var numDecodings = function (s) {
if (s == null || s.length == 0) {
return 0;
}
const dp = Array(s.length + 1).fill(0);
dp[0] = 1;
dp[1] = s[0] !== "0" ? 1 : 0;
for (let i = 2; i < s.length + 1; i++) {
const one = +s.slice(i - 1, i);
const two = +s.slice(i - 2, i);
if (two >= 10 && two <= 26) {
dp[i] = dp[i - 2];
}
if (one >= 1 && one <= 9) {
dp[i] += dp[i - 1];
}
}
return dp[dp.length - 1];
};
class Solution:
def numDecodings(self, s: str) -> int:
@lru_cache(None)
def dp(start):
if start == len(s):
return 1
if start > len(s):
return 0
if s[start] != "0":
if s[start : start + 2] <= "26":
return dp(start + 1) + dp(start + 2)
return dp(start + 1)
return 0
return dp(0)
class Solution {
public:
int numDecodings(string s) {
int pre2 = 0, pre1 = 1;
for (int i = 0; i < s.size() && pre1; ++i) {
int cur = 0;
if (s[i] != '0') cur += pre1;
if (i != 0 && s[i - 1] != '0' && (s[i - 1] - '0') * 10 + s[i] - '0' <= 26)
cur += pre2;
pre2 = pre1;
pre1 = cur;
}
return pre1;
}
};