题目描述
leetcode原题链接:跳跃游戏
给定一个非负整数数组 nums
,你最初位于数组的 第一个下标 。数组中的每个元素代表你在该位置可以跳跃的最大长度。判断你是否能够到达最后一个下标。
举例
示例 1: 输入:nums = [2,3,1,1,4] 输出:true
解释:可以先跳 1 步,从下标 0 到达下标 1, 然后再从下标 1 跳 3 步到达最后一个下标。
示例 2:输入:nums = [3,2,1,0,4] 输出:false
解释:无论怎样,总会到达下标为 3 的位置。但该下标的最大跳跃长度是 0 , 所以永远不可能到达最后一个下标。
解题方法: 贪心算法。扫描数组,用一个变量表示当前扫描能跳的最大位置,每次更新最大能跳到的位置max_pos,当max_pos大于数组的最后一个位置时,说明能够达到最后一个下标。
代码
#include <iostream>
#include <vector>
bool jump_success(const std::vector<int>& nums) {
int n = nums.size();
if (n <= 1) {
return true;
}
int max_pos = 0;
for (int i = 0; i <= max_pos; i++) {
max_pos = std::max(max_pos, i + nums[i]);
if (max_pos >= n - 1) {
return true;
}
}
return false;
}
void print_result(const std::vector<int>&nums, bool result) {
std::cout << "========" << std::endl;
std::cout << "input:" << std::endl;
for (auto num : nums) {
std::cout << num << " ";
}
std::cout << std::endl;
std::cout << "output:" << std::endl;
if (result) {
std::cout << "true" << std::endl;
} else {
std::cout << "false" << std::endl;
}
}
int main()
{
// case1: [2,3,1,1,4] ==> true
std::vector<int> nums = {2, 3, 1, 1, 4};
bool result = jump_success(nums);
print_result(nums, result);
// case2: [3,2,1,0,4] ==> false
nums = {3, 2, 1, 0, 4};
result = jump_success(nums);
print_result(nums, result);
return 0;
}
代码运行结果如下:
========
input:
2 3 1 1 4
output:
true
========
input:
3 2 1 0 4
output:
false