快乐写代码

T91、解码方法

2020-05-28  本文已影响0人  上行彩虹人

一条包含字母 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的情况。已经连续出现0自己返回解码失败。

 public int numDecodings(String s) {
        char[] ss = s.toCharArray();
        if(ss.length<=1)
            if(ss[0]=='0')
                return 0;
            else
                return 1;
        int num[] = new int[ss.length];
        for(int i = 0; i < ss.length; i++)
            num[i] = ss[i] - '0';
        int[] dp = new int[ss.length];
        if(num[0]==0)
            return 0;
        dp[0] = 1;
        if(num[1]==0){
            if(num[0] >= 3)
                return 0;
            dp[1] = 1;
        } 
        else if(num[0] * 10 + num[1] > 26)
            dp[1] = 1;
        else 
            dp[1] = 2;
        for(int i = 2;i < num.length; i++)
            if(num[i]==0 && num[i-1]==0)
                return 0;
            else if(num[i]==0){
                if(num[i-1]<3){
                    dp[i] = dp[i-2];
                    dp[i-1] = dp[i-2];
                }else
                    return 0;
            }
                
            else if(num[i-1]==0)
                dp[i] = dp[i-1];
            else if(num[i-1]*10+num[i]>26)
                dp[i] = dp[i-1];
            else
                dp[i] = dp[i-1] + dp[i-2];
        return dp[dp.length-1];
    }
上一篇下一篇

猜你喜欢

热点阅读