凌云的博客

行胜于言

LeetCode 算法题 45. 跳跃游戏 II

分类:algorithm| 发布时间:2016-10-12 10:35:00


题目

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

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

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

示例:

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

说明:

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

解法 1

  • step 表示步数
  • distance 表示当前步数下最远能走到哪里
  • next 表示下一步最远能走到哪里
class Solution {
public:
    int jump(vector<int>& nums) {
        int step = 0;
        int distance = 0;
        int next = 0;
        for(int i = 0; i < nums.size() - 1; ++i) {
            next = max(next, i + nums[i]);
            if (i == distance) {
                ++step
                distance = next;
            }
        }

        return step;
    }
};

解法 2

从后往前迭代,用 path 保存要走过的点, 添加一个新的点时如果发现这个点能前进到 path 中最后的点之前的点时, 表示最后一个点是多余的,可以将其从 path 中删除。

class Solution {
public:
    int jump(vector<int>& nums) {
        const int n = nums.size();

        vector<int> path;
        path.push_back(n-1);
        for (int i = n-2; i >=0; --i) {
            if (i + nums[i] >= path.back()) {
                int sz = path.size();
                while (sz - 2 >= 0 && i + nums[i] >= path[sz - 2]) {  //can reach one node further
                    path.pop_back();
                    --sz;
                }

                path.push_back(i);
            }
        }

        return path.size() - 1;
    }
};