【DP】牡牛和牝牛

描述

一年一度的展会要来临了，农民约翰想要把N(1 <= N <= 100,000)只奶牛和公牛安排在单独的一行中。 约翰发现最近公牛们非常好斗；假如两只公牛在这一行中靠的太近，他们就会吵架，以至于斗殴，破坏这和谐的环境。约翰非常的足智多谋，他计算出任何两只公牛之间至少要有K (0 <= K < N)只奶牛，这样才能避免斗殴。 约翰希望你帮助他计算一下有多少种安排方法，可避免任何斗殴的的发生。约翰认为每头公牛都是一样的，每头奶牛都是一样的。因而，只要在一些相同的位置上有不同种类的牛，那这就算两种不同的方法。

输入

第一行:两个用空格隔开的数：N和K

输出

第一行：一个单独的数，即约翰可以安排的方法数。考虑到这个数可能很大，你只要输出mod 5,000,011之后的结果就可以了。

第i位置能放公牛的方案数等于第i-k位置能放公牛的方案数加上第i-k位置能放奶牛的方案数（其实两者相加就是第i-k+1位置能放奶牛的方案数）

注意i<k的时候能且只能放一头公牛，需要提前预处理

f[i][0]为i位置放奶牛的方案数，f[i][1]为i位置放公牛的方案数

所以状态转移方程：

f[i][0]=f[i-1][0]+f[i-1][1]

f[i][1]=f[i-k][0]

 1 #include <cstdio>
 2 #define MOD 5000011
 3 int n,k,f[100001][2];
 4 int main()
 5 {
 6     scanf("%d%d",&n,&k);
 7     f[1][0]=f[1][1]=1;
 8     for (int i=1;i<k;i++) f[i][1]=1;
 9     for (int i=2;i<=n;i++)
10     {
11         f[i][0]=(f[i-1][0]+f[i-1][1])%MOD;
12         if (i-k>=1) f[i][1]=f[i-k][0];else f[i][1]=1;
13     }
14     printf("%d",(f[n][0]+f[n][1])%MOD);
15 }

 

转载于:https://www.cnblogs.com/algonote/p/7545683.html