解码方法
一条包含字母 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) 。
代码1:
public int numDecodings(String s) {
if (s.charAt(0) == '0')
return 0;
int[] dp = new int[s.length() + 1];
dp[0] = 1;
dp[1] = 1;
for (int i = 2; i < dp.length; i++) {
if (s.charAt(i - 1) == '0') {
if (s.charAt(i - 2) == '1' || s.charAt(i - 2) == '2') {
dp[i] = dp[i - 2];
} else {
return 0;
}
} else {
if (s.charAt(i - 2) == '1'
|| (s.charAt(i - 2) == '2' && s.charAt(i - 1) > '0' && s.charAt(i - 1) <= '6')) {
dp[i] = dp[i - 1] + dp[i - 2];
} else {
dp[i] = dp[i - 1];
}
}
}
return dp[s.length()];
}
代码2:优化空间
public int numDecodings2(String s) {
if (s.charAt(0) == '0')
return 0;
int pre1=1;//dp[i-1]
int pre2=1;//dp[i-2]
int curr=1;
for(int i=1;i<s.length();i++){
if (s.charAt(i) == '0') {
if (s.charAt(i-1) == '1' || s.charAt(i - 1) == '2') {
curr=pre2;
} else {
return 0;
}
} else {
if (s.charAt(i - 1) == '1'|| (s.charAt(i - 1) == '2' && s.charAt(i) > '0' && s.charAt(i) <= '6')) {
curr=pre1+pre2;
} else {
curr=pre1;
}
}
pre2=pre1;
pre1=curr;
}
return curr;
}