LeetCode刷题之035Python搜索插入位置

原创于 2018-08-12 12:00:59 发布 · 175 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #Python

LeetCode 专栏收录该内容

59 篇文章

订阅专栏

本文介绍了一种算法，用于在一个已排序的数组中查找指定的目标值。如果找到该值，则返回其索引；若未找到，则返回目标值应当被插入的位置。示例包括不同情况下目标值的查找或插入位置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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

示例 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
        """
        if nums[0] >= target:
            return 0
        elif nums[-1] < target:
            return len(nums)
        else:
            for i in range(1, len(nums)):
                if nums[i] == target:
                    return i
                elif nums[i-1] < target and nums[i] > target:
                    return i