【bzoj3398】【Usaco2009 Feb】【Bullcow 牡牛和牝牛】【组合数学】

Description

    约翰要带N(1≤N≤100000)只牛去参加集会里的展示活动，这些牛可以是牡牛，也可以是牝牛．牛们要站成一排．但是牡牛是好斗的，为了避免牡牛闹出乱子，约翰决定任意两只牡牛之间至少要有K(O≤K<N)只牝牛．

    请计算一共有多少种排队的方法．所有牡牛可以看成是相同的，所有牝牛也一样．答案对5000011取模

Input

    一行，输入两个整数N和K.

Output

 

    一个整数，表示排队的方法数．

Sample Input

4 2

Sample Output

6
样例说明
6种方法分别是：牝牝牝牝，牡牝牝牝，牝牡牝牝，牝牝牡牝，牝牝牝牡，牡牝牝牡

题解：果断用组合数学.

枚举放多少头牡牛。然后先从总数中扣去一定要空出的牝牛数.剩下的就是牡牛可以放的位置。

然后直接求组合数即可。

代码：

#include<iostream>
#include<cstdio>
#define P 5000011
using namespace std;
long long ans,n,k;
long long power(long long a,long long b){
	long long ans;
	for(ans=1;b;b>>=1,a=a*a%P) if(b&1) ans=ans*a%P;
	return ans; 
}
long long cal(long long n,long long m){
	long long s1(1),s2(1);
	if (m>n-m) m=n-m;
	for (long long i=1;i<=m;i++){
	   (s1*=(n-i+1))%=P;
	   (s2*=i)%=P;
	}
	return (s1*power(s2,P-2))%P;
}
int main(){
	scanf("%lld%lld",&n,&k);
	ans=1;
	for (long long i=1;i<=n;i++){
	  long long temp;
	  temp=n-(i-1)*k;
	  if (temp>=i) (ans+=cal(temp,i))%=P;
	  else break;
	}
    cout<<ans<<endl;
}