当前位置：首页 > article >正文

LeetCode：2266. 统计打字方案数（DP Java）

article 2025/1/20 10:53:59

2266. 统计打字方案数

题目描述：

实现代码与解析：

动态规划

原理思路：

2266. 统计打字方案数

题目描述：

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' 。

实现代码与解析：

动态规划

class Solution {

    public static int MOD = 1000000007;
    public static int N = 100010;
    public long[] f1 = new long[N];
    public long[] f2 = new long[N];

    public int countTexts(String pressedKeys) {
        f1[0] = f2[0] = 1;
        f1[1] = f2[1] = 1;
        f1[2] = f2[2] = 2;
        f1[3] = f2[3] = 4;
        for (int i = 4; i < N; i++) {
                f1[i] = (f1[i - 1] + f1[i - 2] + f1[i - 3]) % MOD;
                f2[i] = (f2[i - 1] + f2[i - 2] + f2[i - 3] + f2[i - 4]) % MOD;
        }

        long res = 1;
        int cur  = 0;
        for (int i = 0; i < pressedKeys.length(); i++) {
            char c = pressedKeys.charAt(i);
            cur++;
            if (i == pressedKeys.length() - 1 || c != pressedKeys.charAt(i + 1)) {
                res = res * (c != '7' && c != '9' ? f1[cur] : f2[cur]) % MOD;
                cur = 0;
            }
        }
        return (int)res;
    }
}