LeetCode 55題 跳躍遊戲演算法介紹 Python3
阿新 • • 發佈:2018-12-18
先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
如有雷同,純屬巧合,可能不小心撞車了吧。。。