题目来源:https://leetcode.com/problems/jump-game/description/
标记,类似dfs
55. Jump Game
题目大意
从第一个位置开始跳,看能否跳到最后一个位置。
For example:
A = [2,3,1,1,4], return true.
A = [3,2,1,0,4], return false.
思路
类似dfs,标记法,记录步长,根据步长调整visited数组。
优点:可减少扫描次数,做到O(n)。
解题代码
class Solution {
public:
bool canJump(vector<int>& nums) {
int n = nums.size();
if (n == 1) return true;
if (n > 1 && nums[0] == 0) return false;
int visited[n] = {0};
int step = nums[0];
for (int i = 1; i < n; i++) {
if (step > 0) {
visited[i] = 1;
step--;
}
if (step > 0 && step < nums[i]) step = nums[i];
if (step == 0) {
step = nums[i];
}
if (visited[i] == 0) return false;
}
if (visited[n-1] == 1) return true;
else return false;
}
};
根据答案提供的思路,还改了一版贪心算法的解法:
class Solution {
public:
bool canJump(vector<int>& nums) {
int n = nums.size();
int lastPos = n - 1;
for (int i = lastPos; i >= 0; i--) {
if (i + nums[i] >= lastPos) {
lastPos = i;
}
}
return lastPos == 0;
}
};
时间复杂度
O(n)