【leetcode】91 解码方法（动态规划）

题目链接：https://leetcode-cn.com/problems/decode-ways/

题目描述

一条包含字母 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不能单独解码

代码

class Solution {
public:
    int numDecodings(string s) {
        if(s.size() == 0 || (s.size() == 1 && s[0] == '0')) return 0;   // 0不能解码
        if(s.size() == 1) return 1;
        vector<int > dp(s.size() ,0);
        dp[0] = s[0] == '0' ? 0:1;
        for (int i = 1; i < s.size(); ++i) {
            if(s[i] == '0') // 0不能单独解码
                dp[i] = 0;
            else
                dp[i] = dp[i-1];
            int x = s[i-1] - '0';
            int y = s[i] - '0';
            int num = x * 10 + y;
            if(num >= 10 && num <= 26)
                dp[i] += i>1? dp[i-2] :1;
        }
        return dp.back();
    }
};