leetcode之解码方法（C++）

参考链接

1. https://leetcode-cn.com/problems/decode-ways/
2. https://leetcode-cn.com/problems/decode-ways/solution/jie-ma-fang-fa-by-leetcode-solution-p8np/

题目描述

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

要解码已编码的消息，所有数字必须基于上述映射的方法，反向映射回字母（可能有多种方法）。例如，“11106” 可以映射为：

“AAJF” ，将消息分组为 (1 1 10 6)
“KJF” ，将消息分组为 (11 10 6)
注意，消息不能分组为 (1 11 06) ，因为 “06” 不能映射为 “F” ，这是由于 “6” 和 “06” 在映射中并不等价。

给你一个只含数字的 非空 字符串 s ，请计算并返回解码方法的总数。

题目数据保证答案肯定是一个 32 位 的整数。

解题思路

题目等价于“一条消息最多有几种解码方法”，显然是一个“最”型动态规划问题。“状态”就是字符串的索引，“选择”就是将s[i]作为独立编码，还是将s[i-1]s[i]作为整体编码。当s[i]作为独立编码时，需要它不等于’0’，此时只需要考虑s[0:i-1]有多少种就可以了；当s[i-1]s[i]作为整体编码时，需要s[i-1]不等于’0’且s[i-1]s[i]对应的整数小于26。

代码

我的代码

有一些边界处理，可以进一步压缩状态表的大小，因为i位置的状态只与i-1和i-2的状态相关。

class Solution {
public:
    int numDecodings(string s) {
        vector<int> nums(s.size());
        if (s[0] == '0')
        {
            return 0;
        }
        nums[0] = 1;

        if (s.size() > 1)
        {
            if (s[1] != '0')
            {
                nums[1] += 1;
            }
            if ((s[0] - '0') * 10 + (s[1] - '0') <= 26)
            {
                nums[1] += 1;
            }
        }

        for (int i = 2; i < s.size(); i++)
        {
            if (s[i] != '0')
            {
                nums[i] += nums[i - 1];
            }
            if (s[i - 1] != '0' && ((s[i - 1] - '0') * 10 + (s[i] - '0') <= 26))
            {
                nums[i] += nums[i - 2];
            }
        }
        return nums[s.size() - 1];
    }
};

官方代码

边界处理更加凝练，代码更加简洁。

class Solution {
public:
    int numDecodings(string s) {
        int n = s.size();
        vector<int> f(n + 1);
        f[0] = 1;
        for (int i = 1; i <= n; ++i) {
            if (s[i - 1] != '0') {
                f[i] += f[i - 1];
            }
            if (i > 1 && s[i - 2] != '0' && ((s[i - 2] - '0') * 10 + (s[i - 1] - '0') <= 26)) {
                f[i] += f[i - 2];
            }
        }
        return f[n];
    }
};