hdu5642 King's Order(dp)

最新推荐文章于 2018-04-06 15:42:26 发布

原创最新推荐文章于 2018-04-06 15:42:26 发布 · 326 阅读

0 ·

CC 4.0 BY-SA版权

hdu5642

题目

就是说一段小写字母序列，相同的字母不能连续出现超过3次就是合法的，问长度为n的序列有多少种合法的序列。

思路

dp[i][j][k]表示第i位是j字母并且连续了k次，接下来就很简单了。
关键还是要抓住不能超过3次这个特点来列状态方程，谨记。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>

using namespace std;

typedef long long ll;

const int mod=1000000007;
int dp[2010][26][4];

void init()
{
    memset(dp,0,sizeof(dp));
    for(int i=1; i<=26; i++) dp[1][i][1]=1;
    for(int i=2; i<=2000; i++)
    {
        for(int j=1; j<=26; j++)
        {
            dp[i][j][2]=(dp[i][j][2]+dp[i-1][j][1])%mod;
            dp[i][j][3]=(dp[i][j][3]+dp[i-1][j][2])%mod;
            for(int p=1; p<=26; p++)
            {
                if(p!=j)
                    for(int k=1; k<=3; k++)
                        dp[i][j][1]=(dp[i][j][1]+dp[i-1][p][k])%mod;
            }
        }
    }
}

int main()
{
    init();
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        int ans=0;
        for(int i=1; i<=26; i++)
            for(int j=1; j<=3; j++)
                ans=(ans+dp[n][i][j])%mod;
        printf("%d\n",ans);

    }

    return 0;
}