数位dp入门(HDU - 5642 King's Order)

题意：

告诉你国王发布的命令长度为n，由26个小写英文字母组成，命令中不能有连续的长度超过3个的重复的字母，比如aaccc是可以的但acccc是不可以的。问最多有多少种不同的命令是国王发出的，结果对1000000007取模。

dp[i][1]表示长度为i，末尾有j个相同的字母，由题意知，j只能取1，2，3.

dp[i][2]=dp[i-1][1]

dp[i][3]=dp[i-1][2]

dp[i][1]=dp[i-1][1]+dp[i-1][2]+dp[i-1][3];

所以代码为:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<string>
#include<queue>
#include<map>
using namespace std;
const int MOD=1000000007;  
long long int dp[2005][4];//dp[i][j]表示长度为i，末尾有j个相同的字母
                          //dp[i][2]=dp[i-1][1]
                          //dp[i][3]=dp[i-1][2]
                          //dp[i][1]=(dp[i-1][1]+dp[i-1][2]+dp[i-1][3])*25


void init()
{
    memset(dp,0,sizeof(dp));
    dp[1][1]=26;
    for(int i=2;i<2005;i++)
    {
        dp[i][1]=(dp[i-1][1]+dp[i-1][2]+dp[i-1][3])*25%MOD;
        dp[i][2]=dp[i-1][1]%MOD;
        dp[i][3]=dp[i-1][2]%MOD;
    }
 } 

int main()
{
  int T,n;
  cin>>T;
  init();
  while(T--)
  {
    cin>>n;
    int sum;
    sum=(dp[n][1]+dp[n][2]+dp[n][3])%MOD;
    cout<<sum<<endl;
  }
    return 0;
}