leetcode 91. 解码方法(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) 。

解法:

class Solution {
public:
    int numDecodings(string s) {
        int sz = s.size();
        if(sz == 0 || s[0] == '0'){
            return 0;
        }else{
            int pre = 1, cur = 1;
            for(int i = 1; i < sz; i++){
                int tmp = 0;
                if(s[i] != '0'){
                    tmp += cur;
                }
                string num = s.substr(i-1, 2);
                if(num >= "10" && num <= "26"){
                    tmp += pre;
                }
                pre = cur;
                cur = tmp;
                if(pre == 0 && cur == 0){
                    return 0;
                }
            }
            return cur;
        }
    }
};

猜你喜欢

转载自www.cnblogs.com/zhanzq/p/10782426.html