【力扣】91.解码方法

题目描述(中等)

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

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

给定一个只包含数字的非空字符串，请计算解码方法的总数。

题目链接

https://leetcode-cn.com/problems/decode-ways/

示例 1:

输入: "12"
输出: 2
解释: 它可以解码为 "AB"（1 2）或者 "L"（12）。

示例 2:

输入: "226"
输出: 3
解释: 它可以解码为 "BZ" (2 26), "VF" (22 6), 或者 "BBF" (2 2 6) 。

算法分析

设dp[i]表示以第i个字符为结尾时有多少种编码方式，此时i可以单独成码，有dp[i-1]种方法，也可以和前面一个字符一起编码，有dp[i-2]种方法，因此状态转移方程可以写为：

如果出现连续的0, 那么就没有办法分割，返回0。

提交代码：

class Solution {
public:
    int numDecodings(string s) {
        if (s.empty() || s[0] == '0') return 0;
        if (s.size() == 1) return 1;

        const int n = s.size();
        int dp[n];
        memset(dp, 0, sizeof(dp));

        dp[0] = 1;
        if (s[1] != '0') ++dp[1];
        if (s.substr(0, 2) <= "26") ++dp[1];

        for (int i = 2; i < n; ++i) {
            if (s[i] != '0') dp[i] += dp[i - 1];
            if (s[i - 1] == '0' && s[i] == '0') return 0;
            if (s[i - 1] != '0' && s.substr(i-1, 2) <= "26") dp[i] += dp[i - 2];
        }

        return dp[n - 1];
    }
};