1307 牡牛和牝牛（组合计数-递推）

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())