跳跃游戏的游戏规则比较简单,若单纯枚举所有的跳法以判断是否能到达最后一个下标需要的时间复杂度为O(),为此,本文采用贪心策略,从最后一个下标开始逆着向前走,若能跳到第一个元素则表明可以完成跳跃游戏,反之不能。此方法只需遍历一次数组,时间复杂度为O(n)。
示例:
图1 跳跃游戏输入输出示例
代码:
class Solution:def canJump(self, nums):goal = len(nums) - 1for i in range(len(nums) - 1, -1, -1):if i + nums[i] >= goal:goal = iif goal == 0:return Trueelse:return False
解释:
1)goal初始为最后一个下标,其代表需要跳入的目标位置。从最后一个位置开始往前跳,如果前一个位置(i)+前一个位置可以跳的最大长度(nums[i])>=goal,表明可以从前一个位置跳到目标位置。如此一来,即可将目标位置更新为前一个位置,如此往复,直到goal==0,即可以跳到第一个位置,表明完成跳跃游戏,反之不能。时间复杂度由枚举法的O()降至O(n)。