Leetcode(55 跳跃游戏）

55. 跳跃游戏

难度中等562

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

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

判断你是否能够到达最后一个位置。

示例 1:

输入: [2,3,1,1,4]
输出: true
解释: 我们可以先跳 1 步，从位置 0 到达 位置 1, 然后再从位置 1 跳 3 步到达最后一个位置。

示例 2:

输入: [3,2,1,0,4]
输出: false
解释: 无论怎样，你总会到达索引为 3 的位置。但该位置的最大跳跃长度是 0 ， 所以你永远不可能到达最后一个位置。

看到题目后第一个思路就是简单的BFS，写完运行结果超时，代码如下：

class Solution {
    struct Note {
		int index;
		int step;
		Note(int i, int s) {
			index = i;
			step = s;
		}
	};
	queue<Note>store;
	vector<bool>flag;
public:
    bool BFS(vector<int>& nums)
	{
		for (int i = 0; i < nums.size(); ++i) {
			flag.push_back(0);
		}
        store.push(Note(0, nums[0]));
        flag[0] = 1;
		while (!store.empty())
		{
			Note front = store.front();
            if (front.index == nums.size()-1)
				return 1;
			//cout << Res<<endl;
			for (int i = 1; i <= front.step; i++)
			{
				if (!flag[front.index + i])
				{
					flag[front.index + i] = 1;
					if (front.index + i == nums.size()-1)
						return 1;
					store.push(Note(front.index + i, nums[front.index + i]));
				}
			}
			store.pop();
		}
		return 0;
	}
    bool canJump(vector<int>& nums) {
        return BFS(nums);
    }
};

思考后，改变思路，题目实际上就是判断在遇到0的时候能不能跳过去，继续往下走，那么保存一个能够跳跃的最大举例，依次遍历并更新这个值，就能够以O(n)的时间复杂度完成，空间复杂度为O(1);

代码如下：

class Solution {
public:
	bool canJump(vector<int>& nums) {
		int max_step = nums[0];
		if (nums.size() == 1)
			return 1;
		for (int i = 0; i < nums.size(); ++i)
		{
			max_step = max_step - 1 > nums[i] ? max_step - 1 : nums[i];
			if (max_step == 0 && i != nums.size()-1)
				return 0;
		}
		return 1;
	}
};

提交结果如下：