题目: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’ 。
题目分析:
每个按钮可以按不同次数出现不同字母(上限为3、4次)
利用线性DP先对所有可能进行求解
再根据字母的多少进行次数相乘即可
class Solution {
public:
int countTexts(string pressedKeys) {
const int N = 100010;
const int Y = 1e9+7;
int n=pressedKeys.size();
long long dp3[N],dp4[N];
//初始化
dp3[1]=dp4[1]=1,dp3[2]=dp4[2]=2,dp3[3]=dp4[3]=4;
dp3[4]=7,dp4[4]=8;
for(int i=5;i<=n;i++)
{
dp3[i]=dp3[i-1]%Y+dp3[i-2]%Y+dp3[i-3]%Y;
dp4[i]=dp4[i-1]%Y+dp4[i-2]%Y+dp4[i-3]%Y+dp4[i-4]%Y;
}
long long res=1;
int pre=pressedKeys[0]-'0',num=0;
for(int i=0;i<pressedKeys.size();i++)
{
int c=pressedKeys[i]-'0';
if(pre!=c)
{
if(pre==7||pre==9)res=res*dp4[num]%Y;
else res=res*dp3[num]%Y;
pre=c;
num=0;
}
num++;
}
if(pre==7||pre==9)res=res*dp4[num]%Y;
else res=res*dp3[num]%Y;
return res;
}
};