1. 程式人生 > >Leetcode篇:搜尋插入位置

Leetcode篇:搜尋插入位置


@author: ZZQ
@software: PyCharm
@file: searchInsert.py
@time: 2018/11/07 19:20
要求:給定一個排序陣列和一個目標值,在陣列中找到目標值,並返回其索引。如果目標值不存在於陣列中,返回它將會被按順序插入的位置。
你可以假設陣列中無重複元素。
示例 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
        """
        nums_len = len(nums)
        nums.sort()
        if nums[0] >= target:
            return 0
        if nums[nums_len-1] < target:
            return nums_len
        for i in range(1,nums_len):
            if target == nums[i]:
                return i
            else:
                if nums[i-1] < target < nums[i]:
                    return i