1, 题目

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 1:

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

示例 2:

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

示例 3:

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

示例 4:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2, 算法

  1. 搜索第一个小于等于目标值的索引即可
  1. # scala
  2. object Solution {
  3. def searchInsert(nums: Array[Int], target: Int): Int = {
  4. val index = nums.indexWhere(_ >= target)
  5. if (index == -1) nums.length else index
  6. }
  7. }
  1. #python
  2. class Solution:
  3. def searchInsert(self, nums: List[int], target: int) -> int:
  4. i = 0
  5. while i < len(nums) and nums[i] < target:
  6. i += 1
  7. return i