约翰要带 NN 只牛去参加集会里的展示活动,这些牛可以是牡牛,也可以是牝牛。
牛们要站成一排,但是牡牛是好斗的,为了避免牡牛闹出乱子,约翰决定任意两只牡牛之间至少要有 KK 只牝牛。
请计算一共有多少种排队的方法,所有牡牛可以看成是相同的,所有牝牛也一样,答案对 50000115000011 取模。
输入格式
一行,输入两个整数 NN 和 KK。
输出格式
一个整数,表示排队的方法数。
数据范围
1≤N≤1051≤N≤105,
0≤K<N0≤K<N
输入样例:
4 2
输出样例:
6
样例解释
66 种方法分别是:牝牝牝牝,牡牝牝牝,牝牡牝牝,牝牝牡牝,牝牝牝牡,牡牝牝牡。
思路:f[n]代表的是以n为最后一个1的所有可能的种数,而利用递推,加上所有倒数第二个1的位置的可能
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010, mod = 5000011;
int n, k;
int f[N], s[N];
int main(){
cin >> n >> k;
s[0] = f[0] = 1;
for (int i = 1; i <= n; i ++){
f[i] = s[max(i - k - 1, 0)];
s[i] = (s[i - 1] + f[i]) % mod;
}
cout << s[n] << endl;
return 0;
}