2266：统计打字方案数（1856 分）

力扣第 292 场周赛第 3 题

题目

Alice 在给 Bob 用手机打字。数字到字母的 对应 如下图所示。

为了 打出 一个字母，Alice 需要 按 对应字母 i 次，i 是该字母在这个按键上所处的位置。

• 比方说，为了按出字母 's' ，Alice 需要按 '7' 四次。类似的， Alice 需要按 '5' 两次得到字母 'k' 。
• 注意，数字 '0' 和 '1' 不映射到任何字母，所以 Alice 不 使用它们。

但是，由于传输的错误，Bob 没有收到 Alice 打字的字母信息，反而收到了 按键的字符串信息 。

• 比方说，Alice 发出的信息为 "bob" ，Bob 将收到字符串 "2266622" 。

给你一个字符串 pressedKeys ，表示 Bob 收到的字符串，请你返回 Alice 总共可能发出多少种文字信息 。

由于答案可能很大，将它对 109 + 7 取余 后返回。

示例 1：

输入：pressedKeys = "22233"
输出：8
解释：
Alice 可能发出的文字信息包括：
"aaadd", "abdd", "badd", "cdd", "aaae", "abe", "bae" 和 "ce" 。
由于总共有 8 种可能的信息，所以我们返回 8 。

示例 2：

输入：pressedKeys = "222222222222222222222222222222222222"
输出：82876089
解释：
总共有 2082876103 种 Alice 可能发出的文字信息。
由于我们需要将答案对 109 + 7 取余，所以我们返回 2082876103 % (109 + 7) = 82876089 。

提示：

• 1 <= pressedKeys.length <= 105
• pressedKeys 只包含数字 '2' 到 '9' 。

相似问题：

• 0017：电话号码的字母组合
• 0091：解码方法

分析

#1

可以按最后一个字母递推。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

class Solution:
    def countTexts(self, pressedKeys: str) -> int:
        mod = 10**9+7
        n = len(pressedKeys)
        f = [1]+[0]*n
        for i in range(1,n+1):
            c = pressedKeys[i-1]
            for j in range(4 if c in '79' else 3):
                if i-j-1<0 or pressedKeys[i-j-1]!=c:
                    break
                f[i] += f[i-j-1]
                f[i] %= mod
        return f[-1]

621 ms

#2

• 可以把连续相同的字符分为一组
• 将每组的方案数累乘即可
• 具体求每组方案数，注意到只跟长度有关
• 因此可以预处理长度对应的方案数

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16

mod = 10**9+7
ma = 10**5
A = [1]
B = [1]
for _ in range(ma):
    A.append(sum(A[-3:])%mod)
    B.append(sum(B[-4:])%mod)

class Solution:
    def countTexts(self, pressedKeys: str) -> int:
        res = 1
        for c,g in groupby(pressedKeys):
            m = len(list(g))
            res *= B[m] if c in '79' else A[m]
            res %= mod
        return res

96 ms