欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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);
    }
}