简单dp之递推(2)--ZOJ 3747
2017-08-31 14:19
375 查看
题目链接:
ZOj 3747题意:
给n个士兵排队,每个士兵有三种类型G、R、P可选,求至少有m个连续G士兵,最多有k个连续R士兵的排列的种数。题目输入输出:
InputThere are multiple test cases. For each case, there is a line containing 3 integers N (0 < N < 1000000), M (0 < M < 10000) and K (0 < K < 10000), separated by spaces.
Output
One line for each case, you should output the number of ways mod 1000000007.
Sample Input
3 2 2
Sample Output
5
Hint
Denote the Garrison, the Recon Corp and the Military Police as G, R and P. Reasonable arrangements are: GGG, GGR, GGP, RGG, PGG.
思路:
由于又是至多又是至少没办法处理,所以统一转换成至多,对于事件A={至多N个G士兵连续},事件B={至多M-1个G士兵连续},那么A-B={连续人数在M到N中取值,即至少m个连续士兵}。统一之后,设dp[i][j]表示前i个士兵,当第i个士兵是第j种兵(假设G:0,R:1,P:2)时,G至多连续u个、R至多连续v个的排列个数。
令sum=dp[i-1][0]+dp[i-1][1]+dp[i-1][2];对于不作要求的P士兵,它可以由第i-1阶段的三种士兵任意转移:dp[i][2]=sum。
另:
对于G种兵
假设 i<=u 这时第i个位置可以随意放入GRP任意一种,即dp[i][0]=sum;
假设i==u+1时这时要排除1~u 全为G的情况 所以此时的dp[i][0]=sum-1;
假设i>u+1 时这时要排除i-u~i-1全为G的情况 这种状态即为i-u~i-1全为G时i-u-1为P或R的情况,即dp[i][0]=sum-dp[i-u-1][1]-dp[i-u-1][2];
对于R种兵,则与上述G种兵类似。
初始化时,对于第0个位置,整体为1即可,也可理解为对第0个位置放P,并没有什么影响。 dp[0][2]=1,dp[0][0]=dp[0][1]=0; (其实只需要dp[0][2]+dp[0][1]+dp[0][0]=0;即可)
这样,我们令u分别等于N和M-1,v等于K,进行两次递推,得到的结果相减即是答案。
代码:
#include <stdio.h> #include <algorithm> using namespace std; #define LL long long #define mod 1000000007 LL dp[1000009][3]; //G:0 R:1 P:2 LL N,M,K; LL fun(LL u,LL v) { int i,j; dp[0][1]=dp[0][0]=0; dp[0][2]=1; for(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)%mod; else dp[i][0]=(sum-dp[i-u-1][1]-dp[i-u-1][2])%mod; if(i<=v)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 [0]+dp [1]+dp [2])%mod; } int main() { while(scanf("%lld%lld%lld",&N,&M,&K)!=EOF) { LL ans=fun(N,K); ans=((ans-fun(M-1,K))%mod+mod)%mod; //注意减法可能出现负数,这儿坑了好几次,取模的时候要特别处理一下。 printf("%lld\n",ans); } return 0; }
相关文章推荐
- UVA 10328 , ZOJ 3747 (DP 递推)
- 【dp】递推 zoj3747
- dp递推 zoj 3747 Attack on Titans
- zoj 3747 Attack on Titans 递推 计数dp
- 【dp】递推 zoj3747
- zoj 3747 Attack on Titans 带限制条件的计数递推dp
- zoj 3537 三角剖分 简单区间DP
- [递推简单dp]-hdu 2041 超级楼梯
- zoj 2872 Binary Partitions(简单递推)
- zoj1074 To the Max(简单的DP)
- ZOJ 1087 Super Jumping! Jumping! Jumping!(简单DP)
- [蓝桥杯] 算法训练 未名湖边的烦恼(dp递推或简单dfs)
- [递推简单dp]-hdu 2050 折线分割平面
- zoj 3747 (DP)
- ZOJ 3872-Beauty of Array(递推||DP)
- zoj 3201 简单树形dp Tree of Tree
- ZOJ 2136 Longest Ordered Subsequence(简单DP)
- 【题解】 zoj 3747 计数dp 进击的巨人
- Attack on Titans ZOJ - 3747 DP
- zoj 3747 Attack on Titans【递推好题】