DP——hihoCoder 1482

• 题目链接： https://hihocoder.com/problemset/problem/1482

• 题意： 求你计算一个长度为N，且只包括3种字符 ‘O’, ‘L’, ‘A’，其中不能有连续3个L，最多只能有一个A的字符串的数量

• 分析：限制条件是针对字符L和A的数量，那我们就用这个数量来表示状态。$DP[i][j][k]$表示前i个字符包含j个A字符和结尾处连续k个L字符时，构成的字符串的数量，那么转移方程为：

for(int i=0;i<=100000;i++)
{
    for(int j=0;j<2;j++)
    {
        for(int k=0;k<3;k++)
        {
            up(DP[i+1][j][0], DP[i][j][k]);
            if(j!=1) up(DP[i+1][j+1][0],DP[i][j][k]);
            if(k!=2) up(DP[i+1][j][k+1],DP[i][j][k]);
        }
    }
}

• AC代码：

/*************************************************************************
    > File Name: test.cpp
    > Author: Akira 
    > Mail: qaq.febr2.qaq@gmail.com 
 ************************************************************************/

#include<bits/stdc++.h>
typedef long long LL;
typedef unsigned long long ULL;
typedef long double LD;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Sqr(a) ((a)*(a))
using namespace std;

#define MaxN 100001
#define MaxM MaxN*10
#define INF 0x3f3f3f3f
#define PI 3.1415926535897932384626
const int mod = 1E9+7;
const double eps = 1e-6;
#define bug cout<<88888888<<endl;
#define debug(x) cout << #x" = " << x;
inline void up(int &a,int b){
    a+=b;
    if(a>=mod)a-=mod;
}

int DP[MaxN][2][3];
//DP[i][j][k]表示前i个字符有j个A和连续K个L
void init()
{
    DP[0][0][0] = 1;
    for(int i=0;i<=100000;i++)
    {
        for(int j=0;j<2;j++)
        {
            for(int k=0;k<3;k++)
            {
                up(DP[i+1][j][0], DP[i][j][k]);
                if(j!=1) up(DP[i+1][j+1][0],DP[i][j][k]);
                if(k!=2) up(DP[i+1][j][k+1],DP[i][j][k]);
            }
        }
    }
}
int main()
{
    //std::ios::sync_with_stdio(false);
    init();
    int N;
    while(~scanf("%d", &N))
    {
        int ans = 0;
        for(int j=0;j<2;j++)
        {
            for(int k=0;k<3;k++)
            {
                up(ans, DP[N][j][k]);
            }
        }
        printf("%d\n", ans);
    }
    //system("pause");
}