背单词

背单词

链接：https://ac.nowcoder.com/acm/contest/326/B

题目描述

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)

 

输入

2
2 2 2
500 20 30

输出

702
175540856

备注:

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

开三维数组，第一维为当前处理到第几个位置，第二维为重复的是元音还是辅音，第三维为重复的次数

#include<bits/stdc++.h>
using namespace std;
const long long MOD=1e9+7;
 
int T;
long long dp[5005][2][55];
 
int main(){
    std::ios::sync_with_stdio(false);
    cin>>T;
    while(T--){
        memset(dp,0,sizeof(dp));
        dp[1][0][1]=5,dp[1][1][1]=21;
        int n,a,b;
        cin>>n>>a>>b;
        long long ans=0;
        for(int i=2;i<=n;i++){
            for(int j=2;j<=a;j++){
                dp[i][0][j]=(dp[i-1][0][j-1]*5)%MOD;
            }
            for(int j=1;j<=a;j++){
                dp[i][1][1]=(dp[i][1][1]+dp[i-1][0][j]*21)%MOD;
            }
            for(int j=2;j<=b;j++){
                dp[i][1][j]=(dp[i-1][1][j-1]*21)%MOD;
            }
            for(int j=1;j<=b;j++){
                dp[i][0][1]=(dp[i][0][1]+dp[i-1][1][j]*5)%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;
            }
        }
        cout<<ans<<endl;
    }
 
}