洛谷P1192台阶问题 O(n)递推

最新推荐文章于 2020-09-07 10:34:54 发布

勇敢铁铁

最新推荐文章于 2020-09-07 10:34:54 发布

阅读量397

点赞数

分类专栏：题解

本文链接：https://blog.csdn.net/qq_41105401/article/details/82908546

版权

题解专栏收录该内容

28 篇文章 0 订阅

订阅专栏

题目描述

有NN级的台阶，你一开始在底部，每次可以向上迈最多K级台阶（最少1级），问到达第N级台阶有多少种不同方式。

输入输出格式

输入格式：

两个正整数N，K。

输出格式：

一个正整数，为不同方式数，由于答案可能很大，你需要输出ans mod 100003后的结果。

输入输出样例

输入样例#1： 复制

5 2

输出样例#1： 复制

说明

对于20\%20%的数据,有N ≤ 10, K≤3;

对于40\%40%的数据，有N ≤ 1000;

对于100\%100%的数据，有N ≤ 100000,K ≤ 100。

题解：

容易想到O(n*k)的方法，这里给出O(n)的递推方法：

设 f [ i ] 为到第 i 级台阶的方法种数，显然 f [ 0 ] = f [ 1 ] = 1

$\large f[i] = \sum_{j = i - k}^{i - 1}f[j] = f[i-1] + \sum_{j = i-k}^{i-2}f[j] = \sum_{j = i-k-1}^{i-2}f[j] + \sum_{j = i-k}^{i-2}f[j] = 2f[i-1] - f[i-k-1]$

注意 i - k - 1 < 0 时 f [ i ] = 2 * f [ i - 1]

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<iostream>
#include<cstdlib>
using namespace std;
const int maxx = 1e5 + 5;
const int mod = 100003;
int f[maxx];
int n, k;
int main()
{
    memset(f, 0, sizeof(f));
    cin>>n>>k;
    f[0] = f[1] = 1;
    for(int i = 2; i <= n; ++i)
    {
        if(i - k - 1 < 0) { f[i] = 2 * f[i - 1] % mod; }
        else { f[i] = (2 * f[i - 1] - f[i - k - 1] + mod ) % mod; }
    }
    cout<<f[n]<<endl;
    return 0;
}