参考博客:https://blog.csdn.net/cc_again/article/details/24841249
题意:给出长度为n,由字符'G','P','R',组成的字符串,要求至少有m个连续的'G’,至多有k个连续的'R'。问有多少满足条件的字符串。
思路:可以转换成求最多有n个连续的'G',至多有k个连续的'R'的字符串数量sum和最多有m-1个连续的'G',至多有k个连续的'R'的字符串数量ans,那么答案便为sum-ans;
dp[i][j]表示长度为i,以j结尾的字符串数量,j(0=G,1=R,2=P);
当第i个为P的情况很好考虑不会对连续的R和G产生影响,dp[i][2]=dp[i-1][0]+dp[i-1][1]+dp[i-1][2];
当第i个为G时
如果i<=u 时 无论怎么放都不会超过u个连续的G这个限制条件 所以dp[i][0]=dp[i-1][0]+dp[i-1][1]+dp[i-1][2];
如果i=u+1时,要排除前u个都放了G的情况,dp[i][0]=dp[i-1][0]+dp[i-1][1]+dp[i-1][2]-1;
如果i>u+1时,要排除从i-1到i-u位置都放了G的情况,dp[i][0]=dp[i-1][0]+dp[i-1][1]+dp[i-1][2]-dp[i-u-1][1]-dp[i-u-1][2];
当第i个为R时
如果i<=v 时 无论怎么放都不会超过u个连续的G这个限制条件 所以dp[i][1]=dp[i-1][0]+dp[i-1][1]+dp[i-1][2];
如果i=v+1时,要排除前v个都放了G的情况,dp[i][1]=dp[i-1][0]+dp[i-1][1]+dp[i-1][2]-1;
如果i>v+1时,要排除从i-1到i-v位置都放了G的情况,dp[i][1]=dp[i-1][0]+dp[i-1][1]+dp[i-1][2]-dp[i-v-1][0]-dp[i-v-1][2];
#include<bits/stdc++.h>
using namespace std;
#define Maxn 1100000
#define LL long long
#define M 1000000007
LL dp[Maxn][5];
LL n, m, k, u, v;
LL Cal() {
dp[0][0] = 1;
dp[0][1] = 0;
dp[0][2] = 0;
for(int i = 1; i <= n; i++) {
LL sum = (dp[i - 1][0] + dp[i - 1][1] + dp[i - 1][2]) % M;
dp[i][2] = sum;
if(i <= u)
dp[i][0] = sum;
else if(i == u + 1)
dp[i][0] = (sum - 1) % M;
else
dp[i][0] = (sum - dp[i - u - 1][1] - dp[i - u - 1][2]) % M;
if(i <= v)
dp[i][1] = sum;
else if(i == v + 1)
dp[i][1] = (sum - 1) % M;
else
dp[i][1] = (sum - dp[i - v - 1][0] - dp[i - v - 1][2]) % M;
}
return (dp[n][0] + dp[n][1] + dp[n][2]) % M;
}
int main() {
while(~scanf("%lld%lld%lld", &n, &m, &k)) {
LL ans;
u = n, v = k;
ans = Cal();
u = m - 1, v = k;
ans = ((ans - Cal()) % M + M) % M;
printf("%lld\n", ans);
}
return 0;
}