3398: [Usaco2009 Feb]Bullcow 牡牛和牝牛
Time Limit: 1 Sec Memory Limit: 128 MBSubmit: 208 Solved: 140
[ Submit][ Status][ Discuss]
Description
约翰要带N(1≤N≤100000)只牛去参加集会里的展示活动,这些牛可以是牡牛,也可以是牝牛.牛们要站成一排.但是牡牛是好斗的,为了避免牡牛闹出乱子,约翰决定任意两只牡牛之间至少要有K(O≤K<N)只牝牛.
请计算一共有多少种排队的方法.所有牡牛可以看成是相同的,所有牝牛也一样.答案对5000011取模
Input
一行,输入两个整数N和K.
Output
一个整数,表示排队的方法数.
Sample Input
4 2
Sample Output
6
样例说明
6种方法分别是:牝牝牝牝,牡牝牝牝,牝牡牝牝,牝牝牡牝,牝牝牝牡,牡牝牝牡
样例说明
6种方法分别是:牝牝牝牝,牡牝牝牝,牝牡牝牝,牝牝牡牝,牝牝牝牡,牡牝牝牡
HINT
Source
题解:dp+前缀和
f[i]表示在i放置牝牛的方案数,如果是暴力dp的话,需要枚举相隔>=k的牛的位置,然后累加。
用前缀和统计一下,就可以O(1)递推了。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define N 100003
#define p 5000011
using namespace std;
int n,m,k,f[N],sum[N];
int main()
{
scanf("%d%d",&n,&k);
memset(f,0,sizeof(f));
memset(sum,0,sizeof(sum));
f[0]=1; sum[0]=1;
for (int i=1;i<=k;i++)
{
f[i]=1;
sum[i]=(sum[i-1]+f[i])%p;
}
for (int i=k+1;i<=n;i++)
{
f[i]=sum[i-k-1]%p;
sum[i]=(sum[i-1]+f[i])%p;
}
int ans=0;
for (int i=0;i<=n;i++)
ans=(ans+f[i])%p;
cout<<ans<<endl;
}