解码方法
题目描述
一条包含字母 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’
- 若前一个位置是 ‘1’ 或 ‘2’,则可以和前一个数进行合法的组合。dp[n] = dp[n-2]
- 否则无法和前一个进行合法组合,自身也无法单独解码。return 0
-
如果当前位置不是 ‘0’
- 如果前一个位置是 ‘1’ 或 自身小于 ‘6’且前一个位置是 ’2‘ ,则它既可以自身独立解码,又可以和前一个数组合进行解码。dp[n] = dp[n-2] + dp[n-1]
- 否则只能独立解码。dp[n] = dp[n-1]
代码
class Solution {
public:
// vector<string> ans;
// string cur;
// int ret = 0;
int numDecodings(string s) {
if(s.length() == 0)
return 0;
int n = s.length();
vector<int> dp(n, 0);
if(s[0] == '0')
return 0;
dp[0] = 1;
if(n == 1)
return dp[0];
if(s[1] != '0' && (s[0] == '1' || (s[0] == '2' && s[1] <= '6')))
dp[1] = 2;
else if(s[1] == '0' && (s[0] == '2' || s[0] == '1'))
dp[1] = 1;
else if(s[1] == '0')
return 0;
else
dp[1] = 1;
for(int i = 2; i < s.length(); i++){
string cur = "";
if(s[i] != '0' && (s[i-1] == '1' || (s[i-1] == '2' && s[i] <= '6'))){
dp[i] = dp[i-1] + dp[i-2];
}
else if(s[i] == '0' && (s[i-1] == '2' || s[i-1] == '1'))
dp[i] = dp[i-2];
else if(s[i] == '0')
return 0;
else
dp[i] = dp[i-1];
}
return dp[n-1];
}
};