【题解】Leetcode周赛292 统计打字方案书 中等

题目来源： Leetcode周赛292

标签：动态规划

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 。

动态规划：

讲真的，我这个预处理挺笨的…

class Solution {
    public int countTexts(String pressedKeys) {
        long MOD = 1000000007;
        int n = pressedKeys.length();
        long [][] dp = new long[10][100000+1];
        for (int i = 2;i <=6;i++){
            dp[i][1] = 1;
            dp[i][2] = 2;
            dp[i][3] = 4;
        }
        for (int i = 2;i <= 6;i++){
            for (int j = 4;j <= n;j++){
                dp[i][j] = (dp[i][j-1] + dp[i][j-2] + dp[i][j-3]) % MOD;
            }
        }
        dp[7][1] = 1;
        dp[7][2] = 2;
        dp[7][3] = 4;
        dp[7][4] = 8;
        for (int j = 5;j <= n;j++){
            dp[7][j] = (dp[7][j-1] + dp[7][j-2] + dp[7][j-3] + dp[7][j-4]) % MOD;
        }
        dp[8][1] = 1;
        dp[8][2] = 2;
        dp[8][3] = 4;
        for (int j = 4;j <= n;j++){
            dp[8][j] = (dp[8][j-1] + dp[8][j-2] + dp[8][j-3]) % MOD;
        }
        dp[9][1] = 1;
        dp[9][2] = 2;
        dp[9][3] = 4;
        dp[9][4] = 8;
        for (int j = 5;j <= n;j++){
            dp[9][j] = (dp[9][j-1] + dp[9][j-2] + dp[9][j-3] + dp[9][j-4]) % MOD;
        }
        List<Long> cnts = new ArrayList<>();
        int l = 0,r = 0;
        while (r < n){
            char c = pressedKeys.charAt(r);
            while (r < n && pressedKeys.charAt(r) == c){
                r += 1;
            }
            cnts.add(dp[c-'0'][r-l]);
            l = r;
        }
        long res = 1;
        for (int i = 0;i < cnts.size();i++){
            res = (res * cnts.get(i)) % MOD;
        }
        return (int)res;
    }
}