山东财经大学1024程序竞赛——台阶问题(Python）

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

题目描述

看到上楼的时候有很多的台阶，于是你想着这样的问题：
一共有N个台阶，你一开始在0台阶，每次可以向上迈1∼K个台阶，请问到达N级台阶有几种方式呢？

输入描述:

一行，两个正整数N,K

输出描述:

一行，一个正整数表示到达第N级台阶的不同方式数（答案会很大，请对100003取模）

示例1

输入

5 2

输出

8

---

思路

斐波那契数列(k=2)——>找规律(k=3,4...)

首先，我们要理解经典问题“要上N个台阶，从0台阶开始，一次最多上2个台阶，求最多有几种爬法。”逆向思考到：最后一步有2种可能（即最后还差一个台阶或最后还差两个台阶），以这2种可能（即倒数第一个台阶）作为终点则又分别对应2种可能，以Ni（i=n,n-1,n-2...）分别表示能够到达终点最后一个台阶、到达倒数第二个台阶、到达倒数第三个台阶... ...的所有的爬法，注意接下来运用到了递归的思想，Nn=N(n-1)+N(n+2)，N(n-1)=N(n-1-1)+N(n-1-2)，以此类推，可以总结到递归的主体部分，命名函数体为dg，那么主体部分就是dg(N)=dg(N-1)+dg(N-2)，以下是描述递归函数的代码：

def dg(N) :
    if N==1 :
        return 1
    if N==2 :
        return 2
    return dg(N-1)+dg(N-2)

发现规律

根据k=2的情况，可以意识到存在某种规律，通过观察，发现若k=2，则当n>2，有Nn=N(n-1)+N(n-2)；若k=3，则当n>3，有Nn=N(n-1)+N(n-2)+N(n-3）... ...若k，则当n>k，有Nn=N(n-1)+N(n-2)+N(n-3)+... ...N(n-k)然而，无法在递归中加入循环（如果你知道怎么用递归做的话请告诉我！）

 那么我们可以正着算，开一个长度为N的数组a，循环赋值a[i]=sum(a[i-k:i])。

 最后只需要解决前k个Nk的值，即"N=k，且最多可以走k步时，最多的爬法"，观察可得，Nk=2的(k-1)次方个。

 代码附上~

N,k=map(int,input().split())
a=[0]*N
for i in range(k) :
    a[i]=2**i
i+=1
while(i<N) :
    a[i]=sum(a[i-k:i])
    i+=1
print(a[-1]%100003)