1. 程式人生 > >LeetCode 55題 跳躍遊戲演算法介紹 Python3

LeetCode 55題 跳躍遊戲演算法介紹 Python3

先po上原題:

給定一個非負整數陣列,你最初位於陣列的第一個位置。

陣列中的每個元素代表你在該位置可以跳躍的最大長度。

判斷你是否能夠到達最後一個位置。

示例 1:

輸入: [2,3,1,1,4]
輸出: true
解釋: 從位置 0 到 1 跳 1 步, 然後跳 3 步到達最後一個位置。

示例 2:

輸入: [3,2,1,0,4]
輸出: false
解釋: 無論怎樣,你總會到達索引為 3 的位置。但該位置的最大跳躍長度是 0 , 所以你永遠不可能到達最後一個位置。

 對於這樣一個題目,最直覺的演算法應該是這樣的:

將陣列從後往前找零,再尋找這個零可否被跳過,直到找到不可跳過的零,或是檢索完畢。

這個方法有人已經有寫出文章,在此不再贅述。

但因為機緣巧合種種原因,上面的演算法我總是沒能寫出來,所以我又想出了另一個演算法:

主要思想提煉出來,其實就是看兩步走一步的形式。

對於陣列,一開始,我們站在第一個數的位置上,看在當前這個數能跳到的 所有數字 加上 其與當前位置的距離,這樣就可以算出當前位置能跳到的最遠位置,進而選擇下一個跳躍點。

以下是Python3的原始碼。

class Solution:
    def canJump(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        i=0
        while True:
            maxjump =0
            ind=0
            for dx, x in enumerate(nums[ i+1: i+1+nums[i] ]):
                if maxjump <= (dx+1+x):
                    maxjump = dx+1+x
                    ind = dx+1
            i += ind
            if (i)>= len(nums)-1:
                return True
            if maxjump ==0 or nums[i]==0:
                return False

如有雷同,純屬巧合,可能不小心撞車了吧。。。