bzoj 3398: [Usaco2009 Feb]Bullcow 牡牛和牝牛（dp+前缀和）

3398: [Usaco2009 Feb]Bullcow 牡牛和牝牛

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 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种方法分别是：牝牝牝牝，牡牝牝牝，牝牡牝牝，牝牝牡牝，牝牝牝牡，牡牝牝牡

HINT

Source

Silver

[ Submit][ Status][ Discuss]

题解：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;
}