【leetcode刷题】T173-跳跃游戏 II

  • 2019 年 10 月 4 日
  • 筆記

【题目】

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

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

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

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

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

【思路】

可以用动态规划来做,对于所有的j=i,dp[i] = min(dp[j] + 1)

怎么使用贪心算法呢?

使用变量max_i存储目前能够到达的最远位置,那么位置i能够到达最远位置为i+nums[i],时刻更新max_i,并记录最小步数即可。

什么时候更新最小步数呢?

当所处位置i等于前一步能够到达最远的地方时。

就是说,对于[2, 3, 1, 1, 4],首先设置max_i=0,end=0,count=0

i=0时,max_i = max(max_i, i + nums[i]) = 2

前一步(第0步)能够到达最远位置end为0,由于i == end,更新end为max_i,更新count=1

i=1时,max_i = max(max_i, i + nums[i]) = 4

i=2时,max_i=4,由于i == end,更新end为4,更新count=2

继续循环。

【代码】

python版本

class Solution(object):      def jump(self, nums):          """          :type nums: List[int]          :rtype: int          """          if len(nums) == 1:              return 0          count = 0          max_i = 0          end = 0          for i in range(len(nums)):              max_i = max(max_i, i + nums[i])              if i >= end:                  end = max_i                  count += 1              if end >= len(nums) - 1:                  return count

C++版本

class Solution {  public:      int jump(vector<int>& nums) {          if(nums.size() == 1)              return 0;          int count=0, max_i=0, end=0;          for(int i=0; i < nums.size(); i++){              max_i = max(max_i, i+nums[i]);              if(i >= end){                  count++;                  end = max_i;              }              if(end >= nums.size() - 1)                  return count;          }          return 0;      }  };