- Leetcode 3154. Find Number of Ways to Reach the K-th Stair
- 1. 解题思路
- 2. 代码实现
- 题目链接:3154. Find Number of Ways to Reach the K-th Stair
1. 解题思路
这一题思路上就是一个动态规划,我们只需要确定一下运行的终止条件,然后写一下地推函数即可。
显然,由于减一操作不能连续进行,因此,如果某一次jump之后到达的位置大于k+1,此时必然就不可能再到达目标位置k了,我们跳出迭代即可。
2. 代码实现
给出python代码实现如下:
class Solution:def waysToReachStair(self, k: int) -> int:@lru_cache(None)def dp(loc, jump, allow_op1):ans = 0if loc == k:ans += 1if allow_op1 and loc != 0:ans += dp(loc-1, jump, False)if loc + jump <= k+1:ans += dp(loc+jump, jump * 2, True)return ansans = dp(1, 1, True)return ans
提交代码评测得到:耗时116ms,占用内存18.4MB。