题目:
给定一个非负整数数组,你最初位于数组的第一个位置。
数组中的每个元素代表你在该位置可以跳跃的最大长度。
你的目标是使用最少的跳跃次数到达数组的最后一个位置。
示例:
输入: [2,3,1,1,4]
输出: 2
解释: 跳到最后一个位置的最小跳跃数是 2。
从下标为 0 跳到下标为 1 的位置,跳 1 步,然后跳 3 步到达数组的最后一个位置。
说明:
假设你总是可以到达数组的最后一个位置。
思路:贪心算法。
假设当前我们的位置是index
,可以跳3
步,那么我们跳去的位置是3
步中可以跳得更远的位置,以跳一步为例,可以达到的最远距离是:index+1+nums[index+1]
。找到最大值,将index
更新为最大值的索引,步长加1
,继续跳。
最初用的深搜,超时。。。。。
代码:
class Solution(object):
def jump(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
if len(nums) == 1:
return 0
step = 0
index = 0
while index < len(nums):
if index + nums[index] >= len(nums) - 1:
return step + 1
max_pos = index+1
max_num = 0
for i in range(index+1, index+nums[index] + 1):
if max_num < i + nums[i]:
max_num = i + nums[i]
max_pos = i
index = max_pos
step += 1
深搜超时代码:
def jump1(nums):
"""
:type nums: List[int]
:rtype: int
"""
min = 9999999999999999
flag = False
def dfs(time, index):
nonlocal min
nonlocal flag
if index == len(nums) - 1:
if time < min:
min = time
flag = True
return
for i in range(1, nums[index] + 1):
if (flag and (time+1 >= min)) or index+i >= len(nums):
break
dfs(time+1, index+i)
dfs(0, 0)
return min