JAVA程序设计:解码方法 2(LeetCode:639)
程序员文章站
2022-07-05 14:43:00
...
一条包含字母 A-Z 的消息通过以下的方式进行了编码:
'A' -> 1
'B' -> 2
...
'Z' -> 26
除了上述的条件以外,现在加密字符串可以包含字符 '*'了,字符'*'可以被当做1到9当中的任意一个数字。
给定一条包含数字和字符'*'的加密信息,请确定解码方法的总数。
同时,由于结果值可能会相当的大,所以你应当对109 + 7取模。(翻译者标注:此处取模主要是为了防止溢出)
示例 1 :
输入: "*"
输出: 9
解释: 加密的信息可以被解密为: "A", "B", "C", "D", "E", "F", "G", "H", "I".
示例 2 :
输入: "1*"
输出: 9 + 9 = 18(翻译者标注:这里1*可以分解为1,* 或者当做1*来处理,所以结果是9+9=18)
说明 :
输入的字符串长度范围是 [1, 105]。
输入的字符串只会包含字符 '*' 和 数字'0' - '9'。
思路:动态规划题,其实这道题真不难,就是坑点太多。。。定义dp[i]:表示前i个数字能够解出的不同码的数量。
坑点:
1)0要特判,因为0一定要和前一个数字组成两位数才能解码,并且当遇到连续两个0的块就不可能成功解码了,直接返回0即可。
2)注意int会爆掉的问题,如果不过分追求内存的话,建议直接开long类型吧!
class Solution {
private int mod=1000000007;
public int numDecodings(String s) {
int n=s.length();
long[] dp=new long[n];
if(n==0 || s.charAt(0)=='0') return 0;
dp[0]=s.charAt(0)=='*'?9:1;
for(int i=1;i<n;i++) {
if(i==1) {
if(s.charAt(i)=='0') {
if(s.charAt(i-1)>'2' && s.charAt(i-1)<='9' || s.charAt(i-1)=='0')
break;
if(s.charAt(i-1)=='*')
dp[i]=2;
else
dp[i]=1;
continue;
}
if(s.charAt(i)=='*') {
dp[i]=(dp[i-1]*9)%mod;
if(s.charAt(i-1)=='1') dp[i]+=9;
else if(s.charAt(i-1)=='2') dp[i]+=6;
else if(s.charAt(i-1)=='*') dp[i]+=15;
}
else {
if(s.charAt(i-1)=='1' || s.charAt(i-1)=='2' && s.charAt(i)<'7')
dp[i]=dp[i-1]+1;
else if(s.charAt(i-1)=='*')
dp[i]=dp[i-1]+(s.charAt(i)<'7'?2:1);
else
dp[i]=dp[i-1];
}
}
else {
if(s.charAt(i)=='0') {
if(s.charAt(i-1)>'2' && s.charAt(i-1)<='9' || s.charAt(i-1)=='0')
break;
if(s.charAt(i-1)=='*')
dp[i]=dp[i-2]*2%mod;
else
dp[i]=dp[i-2];
continue;
}
if(s.charAt(i)=='*') {
dp[i]=(dp[i]+dp[i-1]*9%mod)%mod;
if(s.charAt(i-1)=='1') dp[i]=(dp[i]+dp[i-2]*9%mod)%mod;
else if(s.charAt(i-1)=='2') dp[i]=(dp[i]+dp[i-2]*6%mod)%mod;
else if(s.charAt(i-1)=='*') dp[i]=(dp[i]+dp[i-2]*15%mod)%mod;
}
else {
dp[i]=dp[i-1];
if(s.charAt(i-1)=='1' || s.charAt(i-1)=='2' && s.charAt(i)<'7')
dp[i]=(dp[i]+dp[i-2])%mod;
else if(s.charAt(i-1)=='*')
dp[i]=(dp[i]+dp[i-2]*(s.charAt(i)<'7'?2:1)%mod)%mod;
}
}
}
return (int)(dp[n-1]%mod);
}
}