这个题一看就让我想到了很久以前做的一个划分区间的问题,最后要的也不是划分结果而是划分方式,这个也是这样的,就是两种角色:选当前一个还是选当前一个加前面一个。
状态变量 ans[i]
:前i个字符能够实现的划分方案数
决策:
ans[i-1]
的方案都能采用,否则为0ans[i-2]
的方案都能采用,否则为0ans[i]
即为上述两种决策结果的加和此外还进行了一个优化,由于每一次计算只跟前两个值有关,将整个ans数组省略,用三个状态变量维护。
class Solution { public: int numDecodings(string s) { int now = 0, pre1 = 1, pre2 = 0; for (int i = 0; i < s.size(); ++i) { // cout << now << " " << pre1 << " " << pre2 << endl; now = 0; if (s[i] != '0') { now += pre1; } if (i > 0 && ((s[i - 1] - 48) * 10 + s[i] - 48 >= 10) && ((s[i - 1] - 48) * 10 + s[i] - 48 <= 26)) { // cout << s[i - 1] << " " << s[i] << endl; now += pre2; } pre2 = pre1; pre1 = now; } return now; } };
Solution 1的Python实现
class Solution: def numDecodings(self, s: str) -> int: pre1, pre2 = 1, 0 for i in range(len(s)): now = 0 if s[i] != '0': now += pre1 if i > 0 and 10 <= int(s[i - 1 : i + 1]) <= 26: now += pre2 pre2, pre1 = pre1, now return now