链接:点击打开链接
题意: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;
}