Leetcode做題日記:34. 在排序陣列中查詢元素的第一個和最後一個位置(PYTHON)
阿新 • • 發佈:2019-01-02
給定一個按照升序排列的整數陣列 nums,和一個目標值 target。找出給定目標值在陣列中的開始位置和結束位置。
你的演算法時間複雜度必須是 O(log n) 級別。
如果陣列中不存在目標值,返回 [-1, -1]。
示例 1:
輸入: nums = [5,7,7,8,8,10], target = 8
輸出: [3,4]
示例 2:
輸入: nums = [5,7,7,8,8,10], target = 6
輸出: [-1,-1]
第一次的程式碼,暴力遍歷,32ms
if target not in nums:
return[-1, -1]
i=0
while True:
if nums[i]==target:
if i==len(nums)-1:
return[i,i]
j=1
while True:
if i+j==len(nums):
return[i,i+j-1]
if nums[i+j]==target:
j=j+1
else:
return [i,i+j-1]
i=i+1
第二次,從兩端找到左右,依然是32ms:
if target not in nums:
return[-1,-1]
i=0
while nums[i]!=target:
i=i+1
j=0
while nums[-1-j]!=target:
j=j+1
return [i,len(nums)-1-j]
最後,我的二分查詢不知道哪裡錯了,總是超時
if target not in nums:
return[-1,-1]
st=0
L=len(nums)
en=L-1
while st<=en:
two=(st+en)//2
if nums[two]<target:
en=two
elif nums[two]>target:
en=two
elif nums[two]==target:
st1=two
en1=two
i=1
j=1
while st-j>=0 and nums[st-j]==target:
st=st-1
j=j+1
while en+i<L and nums[en+i]==target:
en=en+1
i=i+1
return[st,en]