题目链接:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=5170
题意:给n个士兵排队,每个士兵三种G、R、P可选,求至少有m个连续G士兵,最多有k个连续R士兵的排列的种数。
详细解法:http://blog.csdn.net/cc_again/article/details/24841249
注意下取模后为负的情况
#include <cstdio> #include <cstring> #include <cmath> #include <iostream> #include <algorithm> #include <queue> #include <cstdlib> #include <stack> #include <vector> #include <set> #include <map> #define LL long long #define mod 1000000007 #define inf 0x3f3f3f3f #define N 1000010 #define clr(a) (memset(a,0,sizeof(a))) using namespace std; LL dp[N][3]; LL m,n,k; LL solve(LL u,LL v) { 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])%mod; dp[i][2]=sum; if(i<=u)dp[i][0]=sum; else if(i==u+1)dp[i][0]=sum-1; else dp[i][0]=(sum-dp[i-u-1][1]-dp[i-u-1][2]+mod)%mod; if(i<=v)dp[i][1]=sum; else if(i==v+1)dp[i][1]=sum-1; else dp[i][1]=(sum-dp[i-v-1][0]-dp[i-v-1][2]+mod)%mod; } return (dp[n][0]+dp[n][1]+dp[n][2])%mod; } int main() { while(scanf("%lld%lld%lld",&n,&m,&k)>0) { printf("%lld ",((solve(k,n)-solve(k,m-1))%mod+mod)%mod); } }