題目
一條包含字母 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,則必須和前一位結合進行解碼(0無法自己解碼),此時有dp[n-2]種方式(假設前i位有dp(i)種方式)
如果當前位爲1-9,有兩種情況
1、單獨解碼,此時有dp[n-1]種方式
2、和前一位結合解碼,此時有dp[n-2]種方式。(這種解碼方式需要滿足前一位和當前位結合在1-26之間)
所以狀態轉移方程爲dp[i] = dp[i-1] + dp[i-2](滿足一定條件時,不滿足則爲dp[i] = dp[i-1] )
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];
}