题目描述
一条包含字母 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) 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/decode-ways
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
动态规划+斐波那契数列。dp[i]记录前i-1个字符串解码方法的总数。dp[i]的值由s[i-1]和s[i-2]决定:
1.当s[i-1]和s[i-2]都为‘0’;直接返回0即可,因为00无论如何也无法进行正确解析;
2.当s[i-1]和s[i-2]都不为‘0’;如果s[i-1]和s[i-2]组成的字符串对应的数字大于26,说明s[i-1]和s[i-2]不能组合,dp[i]=dp[i-1];否则dp[i]=dp[i-1]+dp[i-2];
3.当s[i-1]不为0和s[i-2]为‘0’;如03,只能产生一种组合,dp[i]=dp[i-2];
4.当s[i-1]为0和s[i-2]不为‘0’;如果s[i-1]和s[i-2]组成的字符串对应的数字大于26,如30,此时0无法解析,直接返回0即可;否则(如20),0只能跟前一个数组合,所以dp[i]=dp[i-2];
代码(c++)
class Solution {
public:
int numDecodings(string s) {
int len=s.length();
if(len==1&&s[0]!='0') return 1;
if(s[0]=='0') return 0;
vector<int> dp(s.length()+1,0);
dp[0]=1;
dp[1]=1;
for(int i=2;i<=len;i++){
string temp;
temp+=s[i-2];temp+=s[i-1];
if(s[i-1]!='0'&&s[i-2]!='0') dp[i]=temp>"26"?dp[i-1]:dp[i-1]+dp[i-2];
else if(s[i-1]=='0'&&s[i-2]!='0'){
if(s[i-2]>'2') return 0;
else dp[i]=dp[i-2];
}
else if(s[i-1]!='0'&&s[i-2]=='0'){
dp[i]=dp[i-2];
}
else return 0;
}
return dp[len];
}
};