HDU 5642 King's Order 动态规划

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5642
题意：
问你长度为n的字符串，只含有小写字母。

没有超过3个连续相同。

问你这个字符串一共有多少种。

解法：

dp[i][j]表示第i个位置，当前连续长度为j的方案数。

然后转移就好了。

可以直接预处理出来。

//HDU 5642

#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9+7;
const int maxn = 2007;
long long dp[maxn][4]; //dp[i][j]表示第i个位置，当前连续长度为j的方案数

int main()
{
    dp[1][1] = 26;
    for(int i = 1; i <= 2000; i++){
        for(int j = 1; j <= 3; j++){
            if(dp[i][j]){
                if(j!=3) dp[i+1][j+1] = (dp[i][j] + dp[i+1][j+1])%mod;//前后可以相同
                dp[i+1][1] = (dp[i+1][1]+dp[i][j]*25)%mod;
            }
        }
    }
    int t, n;
    scanf("%d", &t);
    while(t--){
        scanf("%d", &n);
        long long ans = 0;
        for(int i = 1; i <= 3; i++){
            ans = (ans + dp[n][i]) % mod;
        }
        printf("%lld\n", ans);
    }
    return 0;
}