背单词（DP）

链接：https://ac.nowcoder.com/acm/contest/326/B
来源：牛客网
 

时间限制：C/C++ 2秒，其他语言4秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

winterzz1准备考4级了，现在winterzz1决定把世界上所有单词都背一遍，winterzz1发现任意一个单词最多有A个连续的元音，最多有B个连续的辅音。且单词最长长度为N，winterzz1问你在满打满算的情况他需要背多少单词？？？

输入描述:

首先输入一个T(T<=100)，表示有T组案例，每组案例依次输入三个正整数N,A,B，N<=5000,A<=50,B<=50;

输出描述:

输出winterzz1最多需要背多少单词，结果mod(10^9+7)

示例1

输入

复制

2
2 2 2
500 20 30

输出

复制

702
175540856

备注:

元音字母为a,e,i,o,u，其余21个字母均为辅音

#include<set>
#include<map>
#include<list>
#include<queue>
#include<stack>
#include<math.h>
#include<vector>
#include<bitset>
#include<iomanip>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#define eps (1e-8)
#define MAX 0x3f3f3f3f
#define u_max 1844674407370955161
#define l_max 9223372036854775807
#define i_max 2147483647
#define re register
#define pushup() tree[rt]=max(tree[rt<<1],tree[rt<<1|1])
#define nth(k,n) nth_element(a,a+k,a+n);  // 将 第K大的放在k位
#define ko() for(int i=2;i<=n;i++) s=(s+k)%i // 约瑟夫
#define ok() v.erase(unique(v.begin(),v.end()),v.end()) // 排序,离散化
using namespace std;

inline int read(){
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' & c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}

typedef long long ll;
const double pi = atan(1.)*4.;
const int M=63;
const int N=1e5+5;
int n,a,b;
ll mod=1e9+7;
ll dp[5005][2][55];  // dp[i][0][j] 长度为 i 元音已经连续出现了 j 次
                     // dp[i][1][j] 长度为 i 辅音已经连续出现了 j 次
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d %d %d",&n,&a,&b);
        memset(dp,0,sizeof(dp));
        dp[1][0][1]=5;          // 第一位单词为元音
        dp[1][1][1]=21;         // 第一位单词为辅音
        ll ans=0;
        for(int i=2;i<=n;i++){
            for(int j=2;j<=min(a,i);j++)
                dp[i][0][j]=dp[i-1][0][j-1]*5%mod;
            for(int j=2;j<=min(b,i);j++)
                dp[i][1][j]=dp[i-1][1][j-1]*21%mod;
            for(int j=1;j<=min(b,i-1);j++)
                dp[i][0][1]=(dp[i][0][1]+dp[i-1][1][j]*5)%mod;
            for(int j=1;j<=min(a,i-1);j++)
                dp[i][1][1]=(dp[i][1][1]+dp[i-1][0][j]*21)%mod;
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=a;j++)
                ans=(ans+dp[i][0][j])%mod;
            for(int j=1;j<=b;j++)
                ans=(ans+dp[i][1][j])%mod;
        }
        printf("%lld\n",ans%mod);
    }
    return 0;
}