剑指 offer 学习之解码方法

题目描述

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

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

给定一个只包含数字的非空字符串，请计算解码方法的总数。
题目链接：力扣

解题思路

public class Main {
    
    public static void main(String[] args) {
        String s = "12258";
        System.out.println(numDecodings(s));    // 5 种 分别是 ABBEH，LBEH，AVEH，ABYH，LYH
        
        String s2 = "12";
        System.out.println(numDecodings(s2));    // 2 种 分别是 AB，L
        
        String s3 = "226";
        System.out.println(numDecodings(s3));    // 3 种 分别是 BZ，VF，BBF
    } 
    
    public static int numDecodings(String s) {
        if (s == null || s.length() == 0){
            return 0;
        }
            
        int n = s.length();
        int[] dp = new int[n + 1];
        dp[0] = 1;
        dp[1] = s.charAt(0) == '0' ? 0 : 1;
        for (int i = 2; i <= n; i++) {
            int one = Integer.valueOf(s.substring(i - 1, i));
            if (one != 0) {
                dp[i] += dp[i - 1];
            }
            
            if (s.charAt(i - 2) == '0'){
               continue; 
            }
            
            int two = Integer.valueOf(s.substring(i - 2, i));
            if (two <= 26) {
                dp[i] += dp[i - 2];   
            }
            
        }
        return dp[n];
    }
}

测试结果