Leetcode 45:跳跃游戏 II(最详细的解法!!!)

给定一个非负整数数组,你最初位于数组的第一个位置。

数组中的每个元素代表你在该位置可以跳跃的最大长度。

你的目标是使用最少的跳跃次数到达数组的最后一个位置。

示例:

输入: [2,3,1,1,4]
输出: 2
解释: 跳到最后一个位置的最小跳跃数是 2。
     从下标为 0 跳到下标为 1 的位置,跳 1 步,然后跳 3 步到达数组的最后一个位置。

说明:

假设你总是可以到达数组的最后一个位置。

解题思路

这个问题是之前Leetcode 55:跳跃游戏(最详细的解法!!!)的提升。我们先看看这个问题能不能通过动态规划解决,只是对之前的问题稍加修改即可。

我们可以定义函数 f ( i ) f(i) f(i)表示到达位置 i i i所需的最少步数,那么我们只需要遍历前面的所有位置找到最少步数即可。

  • f ( i ) = m i n ( f ( i ) , f ( j ) + 1 ) f(i)=min(f(i),f(j)+1) f(i)=min(f(i),f(j)+1)

其中 j < i j<i j<i并且 n u m s [ j ] + j > = i nums[j]+j>=i nums[j]+j>=i

class Solution:
    def jump(self, nums: List[int]) -> int:
        nums_len = len(nums)
        mem = [float('inf')]*nums_len
        mem[0] = 0
        for i in range(1,nums_len):
            for j in range(i):
                if nums[j] + j >= i:
                    mem[i] = min(mem[j]+1, mem[i])
                    break
        
        return mem[-1]

但是这样写法超时了,需要更快的解法。这个问题是否可以通过贪心解决呢?

我们将nums按照从大到小排序,也就是前面步子尽量迈大一点,后面就有更多的空间了啊?显然这种做法是不可行的,因为我们每次求解的是局部最优解,而对于全局来说就不一定是最优解了。

我们可以换一个思路,我们先思考跳一步的话最远可以跳多远,接着思考跳两步的话最远可以跳多远,以此类推,直到最远的距离大于等于nums.size()-1,那么此时的步数自然就是最少的步数。

通过pre记录step-1步最多可以跳多远,cur记录step步最多可以跳多远。接着就是思考边界问题,对于每个index来说:

  • pre < index <= cur:那么需要再跳一步。
  • cur >= nums.size()-1:此时必然可以跳到终点,返回步数即可。
class Solution:
    def jump(self, nums: List[int]) -> int:
        n = len(nums) 
        pre, cur, step = -1, 0, 0
        for i in range(n):
            if cur >= n - 1: 
                return step
            elif pre < i <= cur:
                step, pre = step + 1, cur
            cur = max(cur, i + nums[i])     
        return step

非常简洁,非常酷!!!

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题,希望大家指出!!!

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值