题目
一条包含字母 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[i] 表示 s[0 … i - 1] 的解码方法的个数。
状态转移:dp[i + 1] 即为当前 i 位置的数字作为 独立数字 和 个位数字 的解码方法之和。
详细情况如下:
- 独立数字:
- 是 0 :0 不能形成独立数字(编码是从 1 开始的),此种情况编码方法个数为 0。
- 不是 0 : 作为独立数字,那么此种情况编码方法个数取决于 dp[i] 。
- 作为个位数字:如果能作为个位数字,那么此种情况编码方法个数取决于 dp[i - 1] 。
初始化:dp[0] = 1 。
结果:dp[n] 。
复杂度:O(n) 时间,O(n) 空间。
代码
class Solution {
public int numDecodings(String s) {
int n = s.length();
int[] dp = new int[n + 1];
dp[0] = 1;
for(int i = 0; i < n; i ++) {
char cur = s.charAt(i);
dp[i + 1] = cur == '0' ? 0 : dp[i];
if(i > 0 && (s.charAt(i - 1) == '1' || (s.charAt(i - 1) == '2' && (cur - '0') <= 6)))
dp[i + 1] += dp[i - 1];
}
return dp[n];
}
}