Description
Given an array of non-negative integers, you are initially positioned at the first index of the array.
Each element in the array represents your maximum jump length at that position.
Your goal is to reach the last index in the minimum number of jumps.
For example:
Given array A = [2,3,1,1,4]
The minimum number of jumps to reach the last index is 2. (Jump 1 step from index 0 to 1, then 3 steps to the last index.)
Note:
You can assume that you can always reach the last index.
Solution
这道题目与上一道题目的题意基本相同,只不过这道题目是要计算到达最后位置的最短步数。
我们可以采用BFS的方法,记住每一次的移动都可以到达那些位置。
一开始我想记住每一次可以到达的位置,但最后发现会超时,然后结合本题,我们可以每一次只需记录最远可以到达的位置即可,而不需记录每一个位置。
用curMax记录当前可以到达的最远位置,当前位置为i,i必须小于等于curMax;用nextMax记录移动一步后可以到达的最远位置。要记住,当前位置可以是从i到curMax的任意点。
class Solution {
public:
int jump(vector<int>& nums) {
if (nums.size() < 2) return 0;
int result = 0;
int curMax = 0, i = 0, nextMax = 0;
while (i <= curMax) {
result++;
for (;i <= curMax; i++) {
nextMax = max(nextMax, i+nums[i]);
if (nextMax >= nums.size()-1) return result;
}
curMax = nextMax;
}
return 0;
}
};