zoj3747(递推)

链接：点击打开链接

题意：n个士兵排队,每个士兵可选择G,R,P三种字母，求至少有m个连续士兵选G，最多有k个连续士兵选R的排列的种数

代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const long long MOD=1000000007;
long long dp[1000005][3];
long long n,m,k;
long long cal(long long u,long long v){         //dp[i][0]表示表示第i个为G,至多有u个连续G,至多有v个连续R的个数
    long long i,sum;                            //dp[i][1]表示表示第i个为R,至多有u个连续G,至多有v个连续R的个数
    dp[0][0]=1;                                 //dp[i][2]表示表示第i个为P,至多有u个连续G,至多有v个连续R的个数
    dp[0][1]=dp[0][2]=0;
    for(i=1;i<=n;i++){
        sum=(dp[i-1][0]+dp[i-1][1]+dp[i-1][2])%MOD;
        dp[i][2]=sum;                           //当前是P不影响G,R的情况
        if(i<=u)
        dp[i][0]=sum;
        else if(i==u+1)                         //减去连续u+1个G的情况
        dp[i][0]=(sum-1)%MOD;
        else                                    //减去i-1~i-u-1全是G的情况
        dp[i][0]=(sum-dp[i-u-1][1]-dp[i-u-1][2])%MOD;
        if(i<=v)                                //R采用同样的处理方式
        dp[i][1]=sum;
        else if(i==v+1)
        dp[i][1]=(sum-1)%MOD;
        else
        dp[i][1]=(sum-dp[i-v-1][0]-dp[i-v-1][2])%MOD;
    }
    return (dp[n][0]+dp[n][1]+dp[n][2])%MOD;
}
int main(){                                     //因为至少的情况我们不好处理，所以都变成至多
    while(scanf("%lld%lld%lld",&n,&m,&k)!=EOF){ //所以变为至多k个连续R,至多n个连续G-至多k个连续R，至多(m-1)个连续G情况
        printf("%lld\n",((cal(n,k)-cal(m-1,k))%MOD+MOD)%MOD);
    }
    return 0;
}