1. 程式人生 > >Leetcode做題日記:34. 在排序陣列中查詢元素的第一個和最後一個位置(PYTHON)

Leetcode做題日記:34. 在排序陣列中查詢元素的第一個和最後一個位置(PYTHON)

給定一個按照升序排列的整數陣列 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]