1. 问题描述:
约翰要带 N 只牛去参加集会里的展示活动,这些牛可以是牡牛,也可以是牝牛。牛们要站成一排,但是牡牛是好斗的,为了避免牡牛闹出乱子,约翰决定任意两只牡牛之间至少要有 K 只牝牛。请计算一共有多少种排队的方法,所有牡牛可以看成是相同的,所有牝牛也一样,答案对 5000011 取模。
输入格式
一行,输入两个整数 N 和 K。
输出格式
一个整数,表示排队的方法数。
数据范围
1 ≤ N ≤ 10 ^ 5,
0 ≤ K < N
输入样例:
4 2
输出样例:
6
样例解释
6 种方法分别是:牝牝牝牝,牡牝牝牝,牝牡牝牝,牝牝牡牝,牝牝牝牡,牡牝牝牡。
来源:https://www.acwing.com/problem/content/1309/
2. 思路分析:
这道题目属于组合计数的第一大类的题目,一般可以使用递推的方法求解,可以看成是动态规划的一种,我们可以声明一个一维数组f,可以将牡牛看成是1,牝牛看成是0,其中f[i]为所有以长度为i,且以1结尾的字符串的数量,怎么样进行状态的计算呢?因为两个1之间至少要K个0,所以我们可以根据上一个1的位置将其分为若干类,分别对应不同的子集,将这些子集的数量加起来就是当前f[i]对应的字符串数量,并且我们在递归的时候可以维护一个前缀和,这样在计算f[i]的时候就可以直接根据前缀和的值得到当前f[i]的值,最终S[n] = f(1) + f(2) +... f(n)就是答案,也即S[n],初值S[0] = f(0) = 1。
3. 代码如下:
class Solution:
def process(self):
n, k = map(int, input().split())
f, s = [0] * (n + 10), [0] * (n + 10)
f[0] = s[0] = 1
mod = 5000011
for i in range(1, n + 1):
# 当i - k - 1不存在的时候说明前面有若干个0也属于一种方案
f[i] = s[max(0, i - k - 1)]
s[i] = (s[i - 1] + f[i]) % mod
return s[n]
if __name__ == '__main__':
print(Solution().process())