AcWing 1307 牡牛和牝牛

林闻起

已于 2023-07-15 09:14:40 修改

阅读量308

点赞数 1

分类专栏：数学文章标签：算法图论

于 2020-02-03 19:18:49 首次发布

本文链接：https://blog.csdn.net/linwenqidbk/article/details/104160431

版权

数学专栏收录该内容

7 篇文章 0 订阅

订阅专栏

AcWing 1307

题意：

有一个长度为n的01串要求任意两个1之间至少有k个0，求排队数，并对5000011取模。

题解：

设 $f [i]$ 表示以1结尾，长度为i的排列种数。
因此，前一个1必须在0 ~ $i - k - 1$ , (0表示i前面没有1）
所以 $f [i]$ = $\sum_{j=0}^{max(0, i-k-1)}$ $f [j]$
特殊情况 $f [0]$ = 1
答案为 $\sum_{i=0}^{n}$ $f [i]$ ,
如果直接做，时间复杂度为O( $n^2$ ),会超时
因为每次求的都是一个前缀的和，所以可以在求出每个 $f [i]$ 之后计算前缀和.时间复杂度为O(n).
$s [i]$ = $s [i - 1]$ + $f [i]$
$f [i]$ = $s [ma x (0, i - k - 1)]$

代码如下：

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int N = 100005;
const int MOD = 5000011;
int f[N], s[N];
int main(){
	int i, n, k, j, t;
	scanf("%d%d", &n, &k); 
	f[0] = 1;
	s[0] = 1;
	for(i = 1; i <= n; i++){
		t = max(0, i - k - 1);
		f[i] = s[t];
		s[i] = (s[i - 1] + f[i]) % MOD;
	}
	printf("%d\n", s[n]);
	return 0;
}