leetcoe 35 搜尋插入位置
阿新 • • 發佈:2018-12-07
給定一個排序陣列和一個目標值,在陣列中找到目標值,並返回其索引。如果目標值不存在於陣列中,返回它將會被按順序插入的位置。
你可以假設陣列中無重複元素。
示例 1:
輸入: [1,3,5,6], 5
輸出: 2
示例 2:
輸入: [1,3,5,6], 2
輸出: 1
示例 3:
輸入: [1,3,5,6], 7
輸出: 4
示例 4:
輸入: [1,3,5,6], 0
輸出: 0
class Solution(object): def searchInsert(self, nums, target): """ :type nums: List[int] :type target: int :rtype: int """ l = 0 r = len(nums) while l<r: mid = (l+r)//2 if nums[mid]>target: r = mid elif nums[mid]<target: l = mid+1 else: return mid return l
方法二:
class Solution(object):
def searchInsert(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
return len([x for x in nums if x<target])
程式碼三、
class Solution(object): def searchInsert(self, nums, target): """ :type nums: List[int] :type target: int :rtype: int """ if target < nums[0]: return 0 elif target > nums[-1]: return len(nums) r = len(nums) - 1 l = 0 while l<r: mid = (l+r)//2 if nums[mid] == target: return mid elif nums[mid] < target: l = mid + 1 else: r = mid - 1 if nums[l] >= target: return l return l + 1