【Leetcode 每日一题】639. 解码方法 II 【困难】线性DP
一条包含字母 A-Z 的消息通过以下的方式进行了编码:
'A' -> 1
'B' -> 2
...
'Z' -> 26
要 解码 一条已编码的消息,所有的数字都必须分组,然后按原来的编码方案反向映射回字母(可能存在多种方式)。例如,"11106" 可以映射为:
"AAJF" 对应分组 (1 1 10 6)
"KJF" 对应分组 (11 10 6)
注意,像 (1 11 06) 这样的分组是无效的,因为 "06" 不可以映射为 'F' ,因为 "6" 与 "06" 不同。
除了 上面描述的数字字母映射方案,编码消息中可能包含 '' 字符,可以表示从 '1' 到 '9' 的任一数字(不包括 '0')。例如,编码字符串 "1" 可以表示 "11"、"12"、"13"、"14"、"15"、"16"、"17"、"18" 或 "19" 中的任意一条消息。对 "1*" 进行解码,相当于解码该字符串可以表示的任何编码消息。
给你一个字符串 s ,由数字和 '*' 字符组成,返回 解码 该字符串的方法 数目 。
由于答案数目可能非常大,返回对 109 + 7 取余 的结果。
示例 1:
输入:s = "*"
输出:9
解释:这一条编码消息可以表示 "1"、"2"、"3"、"4"、"5"、"6"、"7"、"8" 或 "9" 中的任意一条。
可以分别解码成字符串 "A"、"B"、"C"、"D"、"E"、"F"、"G"、"H" 和 "I" 。
因此,"*" 总共有 9 种解码方法。
示例 2:
输入:s = "1*"
输出:18
解释:这一条编码消息可以表示 "11"、"12"、"13"、"14"、"15"、"16"、"17"、"18" 或 "19" 中的任意一条。
每种消息都可以由 2 种方法解码(例如,"11" 可以解码成 "AA" 或 "K")。
因此,"1*" 共有 9 * 2 = 18 种解码方法。
示例 3:
输入:s = "2*"
输出:15
解释:这一条编码消息可以表示 "21"、"22"、"23"、"24"、"25"、"26"、"27"、"28" 或 "29" 中的任意一条。
"21"、"22"、"23"、"24"、"25" 和 "26" 由 2 种解码方法,但 "27"、"28" 和 "29" 仅有 1 种解码方法。
因此,"2*" 共有 (6 * 2) + (3 * 1) = 12 + 3 = 15 种解码方法。
提示:
1 <= s.length <= 1e5
s[i] 是 0 - 9 中的一位数字或字符 '*'
题解:
比较容易想到这是一个动态规划,而且是线性的,将看成一位数的情况和两位数的情况分别进行讨论,将所有的结果累加,得到最终答案。
class Solution { public: int numDecodings(string s) { int len = s.size(); vector<long long> dp(len + 1); dp[0] = 1; int mod = 1e9 + 7; //一位数的情况 auto check1digit = [](char ch) -> int{ if(ch == '0') return 0; if(ch == '*') return 9; return 1; }; //两位数的情况 auto check2digit = [](char c0, char c1) -> int{ if(c0 == '*' && c1 == '*') return 15; if(c0 == '*'){ if(c1 <= '6') return 2; else return 1; } if(c1 == '*'){ if(c0 == '2') return 6; if(c0 == '1') return 9; return 0; } return c0 != '0' && (c0 - '0') * 10 + (c1 - '0') <= 26; }; for(int i = 1; i <= len; i++){ dp[i] = (long long)dp[i - 1] * check1digit(s[i - 1]) % mod; if(i > 1){ dp[i] = (dp[i] + dp[i - 2] * (long long)(check2digit(s[i - 2], s[i - 1])) % mod); } } return dp[len] % mod; } };