45. Jump Game II

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.

Example:

Input: [2,3,1,1,4]
Output: 2
Explanation: 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.

Solution1:(TLE)

class Solution:
    def jump(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        dp = [100000 for i in range(len(nums))]
        dp[0] = 0
        for i in range(len(nums)):
            for j in range(1,nums[i]+1):
                if i+j<len(nums):
                    dp[i+j] = min(dp[i+j],dp[i]+1)
        return dp[-1]

one case can't pass.

Solution2:

class Solution:
    def jump(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        dp = [100000 for i in range(len(nums))]
        dp[0] = 0
        for i in range(len(nums)):
            if i and nums[i]==nums[i-1]-1:
                continue
            for j in range(1,nums[i]+1):
                if i+j<len(nums):
                    dp[i+j] = min(dp[i+j],dp[i]+1)
        return dp[-1]

A simple optimization.

posted @ 2018-10-16 10:58  bernieloveslife  阅读(92)  评论(0编辑  收藏  举报