leetcode菜狗入门 | 91. 解码方法

解码方法

题目描述

一条包含字母 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’

  • 若前一个位置是 ‘1’ 或 ‘2’，则可以和前一个数进行合法的组合。dp[n] = dp[n-2]
  • 否则无法和前一个进行合法组合，自身也无法单独解码。return 0

• 如果当前位置不是 ‘0’

  • 如果前一个位置是 ‘1’ 或 自身小于 ‘6’且前一个位置是 ’2‘ ，则它既可以自身独立解码，又可以和前一个数组合进行解码。dp[n] = dp[n-2] + dp[n-1]
  • 否则只能独立解码。dp[n] = dp[n-1]

代码

class Solution {
public:
    // vector<string> ans;
    // string cur;
    // int ret = 0;
    int numDecodings(string s) {
        if(s.length() == 0)
            return 0;
        int n = s.length();
        vector<int> dp(n, 0);
        if(s[0] == '0')
            return 0;
        dp[0] = 1;
        if(n == 1)
            return dp[0];
        if(s[1] != '0' && (s[0] == '1' || (s[0] == '2' && s[1] <= '6')))
            dp[1] = 2;
        else if(s[1] == '0' && (s[0] == '2' || s[0] == '1'))
            dp[1] = 1;
        else if(s[1] == '0')
            return 0;
        else
            dp[1] = 1;

        for(int i = 2; i < s.length(); i++){
            string cur = "";
            if(s[i] != '0' && (s[i-1] == '1' || (s[i-1] == '2' && s[i] <= '6'))){
                dp[i] = dp[i-1] + dp[i-2];
            }   
            else if(s[i] == '0' && (s[i-1] == '2' || s[i-1] == '1'))
                dp[i] = dp[i-2];
            else if(s[i] == '0')
                return 0; 
            else 
                dp[i] = dp[i-1];
        }
        return dp[n-1];
    }

};