hdu5642(递推)

链接:点击打开链接

题意:求一个长度是n的并且只由小写字母组成的至多含有三个连续字母的字符串的种数

代码:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const long long MOD=1000000007;
long long n,dp[2005][30];
long long cal(long long u){
    long long i,j,k,sum,tmp;
    sum=0;
    dp[0][0]=1;
    for(i=1;i<26;i++)
    dp[0][i]=0;
    for(i=1;i<=n;i++){
        sum=0;
        for(j=0;j<26;j++)
        sum=(sum+dp[i-1][j])%MOD;
        if(i<=u){
        for(j=0;j<26;j++)
        dp[i][j]=sum;
        }
        else if(i==u+1){
        for(j=0;j<26;j++)
        dp[i][j]=(sum-1)%MOD;
        }
        else{
            for(j=0;j<26;j++){
            tmp=0;
            for(k=0;k<26;k++){
                if(k!=j)
                tmp=(tmp+dp[i-u-1][k])%MOD;
            }
            dp[i][j]=(sum-tmp+MOD)%MOD;
            }
        }
    }
    sum=0;
    for(i=0;i<26;i++)
    sum=(sum+dp[n][i])%MOD;
    return sum;                                 //跟链接上的题类似,只不过变为26个字母
}                                               //http://blog.csdn.net/stay_accept/article/details/51277755
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%I64d",&n);
        printf("%I64d\n",cal(3)%MOD);

    }
    return 0;
}


 

你可能感兴趣的:(hdu5642(递推))