问题描述
小R设计了一款有趣的机器人,它每次移动时至少需要消耗一个能量值。假设当小R为机器人提供了 5 个能量值,它可以选择不动,也可以走 1 步,但它不会走超过 5 步。
小R希望机器人能够走到一个终点,该终点位于 N 个位置的最后一个位置。每个位置上都有一定的能量值,机器人每次消耗当前位置的能量值可以往前走几步,之后可以继续消耗新的位置的能量继续前进。如果某个位置的能量值为 0,机器人将无法再继续行动。
小R想知道,机器人是否有可能通过这些能量值移动到最后一个位置。你能帮他计算一下吗?
测试样例
样例1:
输入:
n = 5 ,array = [2, 3, 1, 1, 4]
输出:'TRUE'
样例2:
输入:
n = 5 ,array = [3, 2, 1, 0, 4]
输出:'FALSE'
样例3:
输入:
n = 6 ,array = [1, 2, 3, 4, 0, 0]
输出:'TRUE'
初步理解
小R设计了一款机器人,每次移动至少消耗一个能量值,提供5个能量值后,机器人可选择不动或最多走5步。机器人需通过消耗各位置上的能量值前进到终点(N个位置的最后一个)。要求判断机器人是否能通过这些能量值到达最后一个位置。
解题思路
-
初始化:设置一个变量
maxReach来记录当前能够到达的最远位置,初始值为 0。 -
遍历数组:从第一个位置开始,遍历每个位置。
- 判断当前位置:如果当前位置
i超过了maxReach,说明无法到达当前位置,返回FALSE。 - 更新最远位置:更新
maxReach为max(maxReach, i + array[i]),即当前位置加上该位置的能量值。 - 判断是否到达终点:如果
maxReach已经超过或等于最后一个位置,返回TRUE。
- 判断当前位置:如果当前位置
-
循环结束:如果遍历完所有位置后还没有到达最后一个位置,返回
FALSE。
关键点
·什么时候能量消耗完
·记录最远位置
代码实现
public class Main {
public static String solution(int n, int[] array) {
// 初始化当前能够到达的最远位置
int maxReach = 0;
// 循环直到到达最后一个位置或者无法前进
for (int i = 0; i < n; i++) {
// 如果当前位置已经超过了能够到达的最远位置,返回 FALSE
if (i > maxReach) {
return "FALSE";
}
// 更新能够到达的最远位置
maxReach = Math.max(maxReach, i + array[i]);
// 如果能够到达最后一个位置,返回 TRUE
if (maxReach >= n - 1) {
return "TRUE";
}
}
// 如果循环结束还没到达最后一个位置,返回 FALSE
return "FALSE";
}
public static void main(String[] args) {
// Add your test cases here
System.out.println(solution(5, new int[] { 2, 3, 1, 1, 4 }).equals("TRUE"));
System.out.println(solution(5, new int[] { 3, 2, 1, 0, 4 }).equals("FALSE"));
}
}
总结
-
问题理解:
- 你需要判断机器人是否能够通过给定的能量值数组到达最后一个位置。每个位置的能量值决定了机器人可以从该位置前进的最大步数。
-
数据结构选择:
- 你可以使用一个变量来记录当前能够到达的最远位置。这个变量可以帮助你判断是否能够继续前进。
·能量决定了最远位置,可能不够也可能太多,所以要记录最远位置,超出最远位置要返回false,循环结束还没到达最后一个位置,返回 FALSE。