与剑指 Offer 46. 把数字翻译成字符串 类似,这个题比那个稍微复杂点
https://leetcode-cn.com/problems/ba-shu-zi-fan-yi-cheng-zi-fu-chuan-lcof/
思路:dp
dp[i]表示以num[i]结尾的解码总数
转移方程:dp[i]=dp[i-1]+dp[i-2]
边界:dp[0] = 1 dp[1] = 1或者2
ans:dp[n-1]
注意:
1:加dp[i-1]的前提:s[i]不为’0’
2:加dp[i-2]的前提:s.substr(i-1, 2)在10-26之间
class Solution {
public:
int numDecodings(string s) {
int n = s.size();
int first = (s[0] != '0'), second;
if (!first) return 0;
if (n > 1) {
int tmp = stoi(s.substr(0, 2));
if (s[1] == '0' && s[0] > '2') second = 0; //当前为‘0’且前一个大于2时,dp[1]=0
else if (s[1] != '0' && tmp >= 10 && tmp <= 26) second = 2;
//当前不为0且s.substr(0, 2)在[10,26],dp[1]=2
else second = 1; //其他情况都为1
}
if (!second) return 0; //
for (int i = 2; i < n; ++i) {
int tmp = stoi(s.substr(i - 1, 2));
int tmp2 = (s[i] != '0' ? second : 0) +
((tmp >= 10 && tmp <= 26) ? first : 0);
if (!tmp2) return 0;
first = second;
second = tmp2;
}
return n > 1 ? second : first;
}
};
易错点:
1:只要存在一个dp[i]为0,就return 0; 包括初始化时的first second。
2:s长度为1的话,应该return first