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 <= 10^5
pressedKeys 只包含数字 ‘2’ 到 ‘9’ 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-number-of-texts
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

很类似91题解码方法，这个题要考虑最后四位，如果按位分析会很麻烦，写出来就像下面的代码一这样，很不优雅。如代码一

可以预先处理出每个字符前面和自己相同的字符个数，这样if语句就可以使用循环合并了。如代码二

代码

代码一

class Solution {
    public int countTexts(String pressedKeys) {
        int mod = (int) (1e9 + 7);
        int n = pressedKeys.length();
        long[] dp = new long[n + 1];
        dp[0] = 1;
        for (int i = 1; i <= n; i++) {
            char c = pressedKeys.charAt(i - 1);
            dp[i] = (dp[i] + dp[i - 1]) % mod;
            if (i >= 2) {
                char c1 = pressedKeys.charAt(i - 2);
                if (c1 == c) {
                    dp[i] = (dp[i] + dp[i - 2]) % mod;
                }
            }
            if (i >= 3) {
                char c1 = pressedKeys.charAt(i - 2);
                char c2 = pressedKeys.charAt(i - 3);
                if (c2 == c && c1 == c) {
                    dp[i] = (dp[i] + dp[i - 3]) % mod;
                }
            }
            if ((c == '7' || c == '9') && i >= 4) {
                char c1 = pressedKeys.charAt(i - 2);
                char c2 = pressedKeys.charAt(i - 3);
                char c3 = pressedKeys.charAt(i - 4);
                if (c3 == c && c2 == c && c1 == c) {
                    dp[i] = (dp[i] + dp[i - 4]) % mod;
                }
            }
        }
        return (int) dp[n];
    }
}

代码二

class Solution {
    public int countTexts(String pressedKeys) {
        int mod = (int) (1e9 + 7);
        int n = pressedKeys.length();
        // f[i]表示第i个字符之前和其相同的字符个数，包含第i个字符本身
        // 如“233355”会得到f:[1,1,2,3,1,2]
        int[] f = new int[n];
        f[0] = 1;
        for (int i = 1; i < n; i++) {
            if (pressedKeys.charAt(i) == pressedKeys.charAt(i - 1)) {
                f[i] = f[i - 1] + 1;
            } else {
                f[i] = 1;
            }
        }
        long[] dp = new long[n + 1];
        dp[0] = 1;
        for (int i = 1; i <= n; i++) {
            char c = pressedKeys.charAt(i - 1);
            // 对于所有数字，均可以考虑最多连续按三次
            // 要注意f[i - 1]可能会大于3，这里和3取min可
            for (int j = 1; j <= Math.min(f[i - 1], 3); j++) {
                dp[i] = (dp[i] + dp[i - j]) % mod;
            }
            // 对于7和9，可以4个连续，但是前提条件是f[i-1]不小于4
            if ((c == '7' || c == '9') && f[i - 1] >= 4) {
                dp[i] = (dp[i] + dp[i - 4]) % mod;
            }
        }
        return (int) dp[n];
    }
}