Python-118：有限制的楼梯攀登

问题描述

小U最近决定挑战一座非常高的楼梯，每次他可以选择走一步或两步，但有一个重要的限制：他不能连续走两步。因此，小U想知道他总共有多少种不同的方式可以从楼梯的底部走到顶端。

你需要帮他计算在给定的楼梯层数下，小U有多少种走法。

代码

def solution(n):

    # Edit your code here

    f = [[1, 0]]

    for i in range(1, n + 1):

        x = f[i - 1][0] + f[i - 1][1]

        y = f[i - 2][0] if i >= 2 else 0

        f.append([x, y])

    return sum(f[-1])

 

if __name__ == "__main__":

    # Add your test cases here

    print(solution(2) == 2)