给定一个非负整数数组,你最初位于数组的第一个位置。
数组中的每个元素代表你在该位置可以跳跃的最大长度。
你的目标是使用最少的跳跃次数到达数组的最后一个位置。
示例:
输入: [2,3,1,1,4]
输出: 2
解释: 跳到最后一个位置的最小跳跃数是 2。
从下标为 0 跳到下标为 1 的位置,跳 1 步,然后跳 3 步到达数组的最后一个位置。
说明:
假设你总是可以到达数组的最后一个位置。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/jump-game-ii
解:
def jump(self, nums: List[int]) -> int:
if len(nums) <= 1:
return 0
position = 0
count = 0
l = len(nums) - 1
while position <= l:
if position + nums[position] >= l:
return count + 1
if nums[position] == 1:
position += 1
count += 1
continue
n = nums[position+1:position+1+nums[position]]
m = max([nums[position+i] + i for i in range(nums[position], 0, -1)])
for i in range(nums[position], 0, -1):
if position + i > l:
continue
if nums[position+i] + position >= l:
return count + 2
elif nums[position+i] + i >= l:
return count + 2
elif nums[position+i] + i == m:
position += i
count += 1
break
return count