BestCoder Round #75 1003 数位DP

King's Order

Accepts: 381

Submissions: 1361

Time Limit: 2000/1000 MS (Java/Others)

Memory Limit: 65536/65536 K (Java/Others)

问题描述

国王演讲后士气大增，但此时战争还没有结束，国王时不时要下发命令。

由于国王的口吃并没有治愈，所以传令中可能出现：“让第三军-军-军，到前线去” 这样的命令。由于大洋国在军队中安插了间谍 ， 战事紧急，很多时候前线的指挥官不能分清哪些命令真正来自国王。但国王的命令有一个特点，他每次连续重复的字符最多 33 次. 所以说他的命令中没有：“让第三军-军-军-军 ， 到前线去”，但是可以有 ：“让第三军-军 ， 到前线去” 。

此时将军找到了你，你需要告诉他，给定命令的长度长度为 nn，有多少种不同的命令可以是国王发出的 。（也就是求长度为 nn 的合格字符串的个数）当然，国王可能说出一句话没有犯任何口吃，就像他那次演讲一样。

为了简化答案，国王的命令中只含有小写英文字母，且对答案输出模 10000000071000000007。

我们认为两个命令如果完全相同那么这两个字符串逐个比较就完全相同。

输入描述

第一行一个整数表示测试组数：T(T \le10)T(T≤10)。

每组数据占一行，每行一个正整数 n(n \le 2000)n(n≤2000) 表示字符串的长度。

输出描述

共 TT 行，每行一个整数表示合法的命令数量。

输入样例

2
2
4

输出样例

676
456950

Hint

两个中没有不符合要求的，所以答案为 26\times 26 = 67626×26=676

四个不符合要求的只有 `aaaa` `bbbb` ... `zzzz`总共 26 个

那么答案就是： 26^4-26 = 45695026​4​​−26=456950

Statistic | Submit | Clarifications | Back

题解：定义DP[i][j][k]表示为i个字符 , 结尾字符为 'a'+j​ , 结尾部分已重复出现了k次的字符串数；如果当前的字符等于前一个字符记为g，则dp[i][j][k] += dp[i-1][g][k-1] ；否则dp[i][j][1] += dp[i-1][g][k];

代码：

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;
typedef long long ll;
#define mod 1000000007
ll dp[2222][30][5];//dp[i][j][k]表示为i个字符 , 结尾字符为 'a'+j​ , 结尾部分已重复出现了k次的字符串数
int main()
{
#ifdef CDZSC_June
freopen("t.txt","r",stdin);
#endif //CDZSC_June
int t,n;
scanf("%d",&t);
while(t--)
{
memset(dp,0,sizeof(dp));
scanf("%d",&n);
for(int i = 0; i<26; i++)
{
dp[1][i][1] = 1;
}
for(int i =1;i<=n; i++) //字符长度的枚举
{
for(int g =0; g<26; g++)//枚举第i位字符‘a’+0~‘a’+25
{
for(int k =1; k<=3; k++)//保证至多只有3个连续重复
{
for(int j =0; j<26; j++)//枚举第i位字符‘a’+0~‘a’+25
{
if(j == g){
dp[i][j][k] += (dp[i-1][g][k-1]%mod);
dp[i][j][k] %= mod;
}
else{
dp[i][j][1] += (dp[i-1][g][k]%mod);
dp[i][j][1] %= mod;
}
}
}
}
}
ll sum = 0;
//统计长度为n而且满足条件的字符数
for(int j =0; j<26; j++)
{
for(int k =1; k<=3; k++)
{
sum += dp
[j][k]%mod;
sum %= mod;
}
}
printf("%lld\n",sum);
}
return 0;
}