639. 解码方法 II

299 阅读3分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动

639. 解码方法 II

一条包含字母 A-Z 的消息通过以下的方式进行了编码:

'A' -> 1
'B' -> 2
...
'Z' -> 26

要 解码 一条已编码的消息,所有的数字都必须分组,然后按原来的编码方案反向映射回字母(可能存在多种方式)。例如,"11106" 可以映射为:

"AAJF" 对应分组 (1 1 10 6) "KJF" 对应分组 (11 10 6) 注意,像 (1 11 06) 这样的分组是无效的,因为 "06" 不可以映射为 'F' ,因为 "6" 与 "06" 不同。

除了 上面描述的数字字母映射方案,编码消息中可能包含 '' 字符,可以表示从 '1' 到 '9' 的任一数字(不包括 '0')。例如,编码字符串 "1" 可以表示 "11"、"12"、"13"、"14"、"15"、"16"、"17"、"18" 或 "19" 中的任意一条消息。对 "1*" 进行解码,相当于解码该字符串可以表示的任何编码消息。

给你一个字符串 s ,由数字和 '*' 字符组成,返回 解码 该字符串的方法 数目 。

由于答案数目可能非常大,返回对 109 + 7 取余 的结果。

示例 1:

输入:s = "*"
输出:9
解释:这一条编码消息可以表示 "1""2""3""4""5""6""7""8""9" 中的任意一条。
可以分别解码成字符串 "A""B""C""D""E""F""G""H""I" 。
因此,"*" 总共有 9 种解码方法。
示例 2:

输入:s = "1*"
输出:18
解释:这一条编码消息可以表示 "11""12""13""14""15""16""17""18""19" 中的任意一条。
每种消息都可以由 2 种方法解码(例如,"11" 可以解码成 "AA""K")。
因此,"1*" 共有 9 * 2 = 18 种解码方法。
示例 3:

输入:s = "2*"
输出:15
解释:这一条编码消息可以表示 "21""22""23""24""25""26""27""28""29" 中的任意一条。
"21""22""23""24""25""26"2 种解码方法,但 "27""28""29" 仅有 1 种解码方法。
因此,"2*" 共有 (6 * 2) + (3 * 1) = 12 + 3 = 15 种解码方法。

数组含义

dp[i]代表字符串前i个字符的解码方法数目

初始化

dp[0]=1。代表空字符串的解码方法只有一种

状态转移

  1. 当只解码一个字符时,将当前字符可能的情况数目与前面的dp[i-1]相乘
  • 字符为*,可以解码为1-9,共9种情况
  • 字符不为*,解法已经确定了,只能有一种解码方法
  1. 解码两个字符时,,将当前字符可能的情况数目与前面的dp[i-2]相乘
  • 字符为**,能解码出11-19 21-26共15种
  • 字符为*c,*只能解码为1或者2,当c小于6时,可以为1或者2(2种),否则只能解码为1(1种)
  • 字符为c*,c只能为1或者2才能进行解码,1可能匹配9种情况,2能匹配6种情况
  • 字符为cc,要判断cc是否处于10-26这个区间(1种情况),否则0种

代码

class Solution {
    public int numDecodings(String s) {
        
        
        
        int n=s.length(),mod=1000000007;
        long[] dp=new long[n+1];
        dp[0]=1;
        for(int i=0;i<n;i++)
        {
            dp[i+1]=dp[i]*getOne(s.charAt(i))%mod;
            if(i>0){
                dp[i+1]=(dp[i+1]+(dp[i-1]*getTwo(s.charAt(i-1),s.charAt(i))))%mod;                
            }
            
        }
        return (int)dp[n];
        
    }
    public int getOne(char c)
    {
        if(c=='0') return 0;
        return c=='*'?9:1;
    }
        public int getTwo(char c1,char c2)
    {
             if(c1=='*'&&c2=='*')
                 return 15;
            if(c1=='*')
                return c2>'6'?1:2;
            if(c2=='*')
            {
                
                if(c1=='2')
                    return 6;
                else if(c1=='1')
                return 9;
                else 
                return 0;
            }
            if(c1=='1'||c1=='2'&&c2<='6')
                return 1;
            return 0;          
    }
}