【LeetCode】91. 解码方法——dp

题目

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

解答

f(n)表示字符串0-n的解码方法总数，若s[n-1]和s[n]能构成解码，则f(n)=f(n-1)+f(n-2)；否则，f(n)=f(n-1)。
注意考虑当前字符为’0’时，判断前一个字符的取值。

class Solution {
public:
    int numDecodings(string s) {
        // f(n) = f(n-1) + f(n-2) (if [s[n-1],s[n]] is valid)
        // f(n) = f(n-1) (else)
        if (s[0] == '0') return 0;
        if (s.length() == 1) return 1;
        vector<int> dp(2, 0);
        dp[0] = 1;
        if (s[1] == '0') {
            if (s[0] > '2') return 0;
            dp[1] = 1;
        }
        else
            dp[1] = (s[0]<'2' || (s[0]=='2' && s[1] <= '6')) ? 2 : 1;

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

            dp[0] = dp[1];
            dp[1] = temp;
        }
        return dp[1];
    }
};