最开始是想用递归来做(作为新手技术没到位,始终AC不了,花了好多时间,无语)
后来想到用下面这个方法来做
先把题目贴过来:
首先有两个规律得先找到:
k=2 : 1 2 3 5 8 13 21 34...
k=3 : 1 2 4 7 13 24 44 81...
k=4 : 1 2 4 8 15 29 56 108...
k=5 : 1 2 4 8 16 31 61 120...
1.根据上述数据可以知道前k项是成等比数列的
2.而后面的与前面的关系为:第n项=第(n-1)项x2-第(n-1-k)项
第一项始终为1,但会发现比如当k=2时,第三项=第二项x2-第零项,若仍然满足上述规律则需要令第零项同样为1.
以下是代码:
#台阶问题
N,K = map(int,input().split())
a = [x for x in range(100001)] #设置一个列表a
a[0]=1
a[1]=1
for i in range(2,N+1): #前两项已经赋值为1,所有从索引2开始遍历
if i <= K: #前k项
a[i] = a[i-1]*2 % 100003 #等比数列公比为2,100003是根据题目要求
else:
a[i] = (a[i-1]*2-a[i-1-K]) % 100003 #第二个规律
ans = a[N]
print(ans)