leetcode--搜索插入位置--python

题目

题目详情

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

你可以假设数组中无重复元素。

示例

输入: [1,3,5,6], 5
输出: 2

输入: [1,3,5,6], 2
输出: 1

输入: [1,3,5,6], 7
输出: 4

输入: [1,3,5,6], 0
输出: 0

解题思路

思路

1. 这道数组题目很简单，只需要遍历再进行判断即可
2. 刚开始我需要写三个判断语句，之后发现可以简化

代码

class Solution(object):
    def searchInsert(self, nums, target):
        for i in range(len(nums)):  #遍历
            if nums[i] >= target:   #判断
                return i
        return i+1    #不满足的判断的处理方法

运行结果

最佳方案

我自己这种方法不知道为什么花了这么长的时间，很疑惑，最佳方案跟我是同一种思想啊，他用了32ms

class Solution:
    def searchInsert(self, nums: 'List[int]', target: 'int') -> 'int':
        try:
            return nums.index(target)
        except:
            for i in range(0,len(nums)):
                if nums[i] > target:
                    return i
            return len(nums)