【ACWing】1307. 牡牛和牝牛

题目地址：

https://www.acwing.com/problem/content/1309/

约翰要带$N$只牛去参加集会里的展示活动，这些牛可以是牡牛，也可以是牝牛。牛们要站成一排，但是牡牛是好斗的，为了避免牡牛闹出乱子，约翰决定任意两只牡牛之间至少要有$K$只牝牛。请计算一共有多少种排队的方法，所有牡牛可以看成是相同的，所有牝牛也一样，答案对$5000011$取模。

输入格式：
一行，输入两个整数$N$和$K$。

输出格式：
一个整数，表示排队的方法数。

数据范围：
$1\le N\le 10^5$
$0\le K<N$

其实就是问长$N$的$01$序列里若使得相邻的$1$之间必须至少有$K$个$0$，有多少种方案。设$f[n]$是以$1$结尾的长$n$的满足条件的序列有多少种，那么我们可以按倒数第$2$个$1$的位置在哪里来分类，所以$f[n]=f[n-K-1]+f[n-K-2]+...+f[0]$。如果$n<K+1$，则$f[n]=1$。最后求总方案的时候则按最后的$1$的出现的位置来分类，所以总方案数量即为$\sum f$。代码如下：

#include <iostream>
using namespace std;

const int N = 1e5 + 10, P = 5000011;
int n, k;
int f[N];

int main() {
  scanf("%d%d", &n, &k);
  for (int i = 0, s = 0; i <= n; i++) {
    if (i < k + 1) f[i] = 1;
    else {
      s = (s + f[i - k - 1]) % P;
      f[i] = s;
    }
  }

  int res = 0;
  for (int i = 0; i <= n; i++) res = (res + f[i]) % P;
  printf("%d\n", res);
}

时空复杂度$O(N)$。