思路:(动态规划)
1.由题意可知,在台阶1时步数为零,在台阶2时,步数为1和2的公倍数2,所以在台阶2时步数是2,我们根据动态规划的思想,我们设立一个列表记录在台阶i时所需要的最小步数。
2.假想一下当你站在台阶i时你可能是从台阶i - j 上过来的(for j in range(1,22)注意一个前提,i-j 是要大于等于0 的),所以台阶i 上最少步数为min(
[(min_step[i-j] + com(i+1,i-j+1)) for j in range(1,22) if i - j >= 0]
) ,直到算出第2021个台阶的最少步数就是我们的答案了。
def com(a:int,b:int):#输出最小公倍数
i = 1
max_ = max(a,b)
min_ = min(a,b)
while i:
if (max_*i) % min_ == 0:
return (max_*i)
i += 1
min_step = [0,2]
for i in range(2,2021):
a = min([(min_step[i-j] + com(i+1,i-j+1)) for j in range(1,22) if i - j >= 0])
min_step.append(a)
print(min_step[2020])