力扣91.解码方法
题目
一条包含字母 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) 。
问题分析
如果当前位为0,则必须和前一位结合进行解码(0无法自己解码),此时有dp[n-2]种方式(假设前i位有dp(i)种方式)
如果当前位为1-9,有两种情况
1、单独解码,此时有dp[n-1]种方式
2、和前一位结合解码,此时有dp[n-2]种方式。(这种解码方式需要满足前一位和当前位结合在1-26之间)
所以状态转移方程为dp[i] = dp[i-1] + dp[i-2](满足一定条件时,不满足则为dp[i] = dp[i-1] )
代码如下:
public int numDecodings(String s) {
int len = s.length();
if(s.charAt(0) == '0') return 0;
if(len == 0 || len == 1) return len;
for(int i = 1;i < len;i++){
//只要有00连在一起就是0,比如100
if(s.charAt(i-1) == '0' && s.charAt(i) == '0'){
return 0;
}
//0前面接着比2大的数就是0,比如50
if(s.charAt(i) == '0' && s.charAt(i-1) > '2'){
return 0;
}
}
int[] dp = new int[len];
dp[0] = 1;
int PreTwoNum = (s.charAt(0) - 48) * 10 + (s.charAt(1) - 48);
//第二位不为0且前两位组合起来处于10-26之间,并且不为20为两种方式,否则为1种方式
dp[1] = PreTwoNum > 10 && PreTwoNum <= 26 && PreTwoNum != 20 ? 2 : 1;
for(int i = 2;i < len;i++){
if(s.charAt(i) == '0'){
dp[i] = dp[i - 2];
}else{
int TwoNum = (s.charAt(i-1) - 48) * 10 + (s.charAt(i) - 48);
if(TwoNum <= 26 && TwoNum >= 11){
dp[i] = dp[i - 1]+dp[i - 2];
}else{
dp[i] = dp[i - 1];
}
}
}
return dp[len - 1];