leetcode91 解码方法(Medium)
题目来源:leetcode91 解码方法
题目描述:
一条包含字母 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) 。
解题思路:
动态规划。dp[i]为0~i的译码总数。
①若s[i]=0,若s[i-1]=1或者2,dp[i]=dp[i-2],否则出错,返回false
②若s[i-1]=1,那么dp[i]可以有两种情况,s[i]和s[i-1]分开译码或者不分开,dp[i]=dp[i-1]+dp[i-2]
③若s[i-2]=2,若s[i]在1~6直接,则可以合并译码dp[i]=dp[i-1]+dp[i-2],否则只能分开译码 dp[i]=dp[i-1]
dp[i]只和前两个变量有关系,可以用两个变量代替。
class Solution {
public:
int numDecodings(string s) {
int n=s.size();
if(n==0) return 0;
if(s[0]=='0') return 0;
int pre=1,cur=1;
for(int i=1;i<n;i++){
int temp=cur;
if(s[i]=='0'){
if (s[i-1]=='1'||s[i-1]=='2')
cur=pre;
else return 0;
}
else if(s[i-1]=='1'||(s[i-1]=='2'&&s[i]>='1'&&s[i]<='6'))
cur=cur+pre;
pre=temp;
}
return cur;
}
};