BC - King's Order - 数位DP

题意：给你一个字符串的长度n在这个n长的字符串里不能出现连续超过3次的的字符。问符合该条件的字符串有几个。

思路：

一个比较暴力的dp方法：

开一个三维dp数组dp[串长度][添加字符前的最后一个字符][最后一个字符重复数]

这样我们对于每个长度枚举新添加的一个字符和倒数第二个，如果新加字符与倒数第二个字符相同我们对于每一个重复情况都向下递推累加。如果不同我们让dp[当前长度][最后一个字符][重复数为1] 累加倒数第二个字符所有重复数情况。最后对于dp[n][i][j]（0<=i<=25,1<=j<=3）求和得解；

一个很巧妙的dp：

其实之前种dp中中间状态可以不用考虑，我们使用一个二维数组让每次长度的递推包含了对应增加的新字符。对于每一次长度增加枚举最后一个字符的重复状态，对于重复1和2次我们可以选择添加一个同样字符所以dp[i+1][j+1] = dp[i+1][j+1]+dp[i][j] %MOD。于此同时新加字符也可能与当前最后一个字符不同所以dp[i+1][1] = dp[i+1][1] + dp[i][j]*25%MOD两个递推在一次枚举时同时进行 。这样会极大的优化dp过程。

两种方法分别如下：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#define MOD 1000000007
using namespace std;

long long dp[2010][27][4];
int main()
{
    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 j=0;j<26;j++)
            {
                for(int k=0;k<26;k++)
                {
                    if(j==k)
                        for(int l=1;l<=3;l++)
                        {
                            dp[i][j][l] = (dp[i][j][l]+dp[i-1][j][l-1]) % MOD;
                        }
                    else
                        for(int l=1;l<=3;l++)
                        {
                            dp[i][k][1] = (dp[i][k][1]+dp[i-1][j][l]) % MOD;
                        }
                }
            }
        }
        long long ans=0;
        for(int i=0;i<26;i++)
        {
            for(int j=1;j<=3;j++)
                ans = (ans+dp[n][i][j]) %MOD;
        }
        printf("%lld\n",ans);
    }
    return 0;
}

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#define MOD 1000000007
using namespace std;

long long dp[2010][4];
int main()
{
    int t,n;
    scanf("%d",&t);
    while(t--)
    {
        memset(dp,0,sizeof(dp));
        scanf("%d",&n);
        dp[1][1]=26;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=3;j++)
            {
                if(j!=3)
                    dp[i+1][j+1] = (dp[i+1][j+1]+dp[i][j]) % MOD;
                dp[i+1][1] = (dp[i+1][1]+dp[i][j]*25) % MOD;
            }
        }
        long long ans=0;
        for(int i=1;i<=3;i++)
        {
            ans = (ans+dp[n][i]) % MOD;
        }
        printf("%lld\n",ans);
    }
    return 0;
}