Binary Search - 35. Search Insert Position

ppxai

发布于 2020-09-23 17:41:43

44600

代码可运行

文章被收录于专栏：皮皮星球皮皮星球

运行总次数：0

代码可运行

35. Search Insert Position

Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.

You may assume no duplicates in the array.

Example 1:

Input: [1,3,5,6], 5
Output: 2

Example 2:

Input: [1,3,5,6], 2
Output: 1

Example 3:

Input: [1,3,5,6], 7
Output: 4

Example 4:

Input: [1,3,5,6], 0
Output: 0

思路：

题目意思是给定一个有序数组，和一个target，找出target在数组中的下标，或者应该插入的位置。使用binary search来做。

代码：

go：

func searchInsert(nums []int, target int) int {

    var (
        start = 0
        end = len(nums) - 1
        mid = 0
    )
    for start + 1 < end {
        mid = (end - start) /2 + start
        if target == nums[mid] {
            return mid
        } else if target < nums[mid] {
           end = mid 
        } else {
            start = mid
        }
    }
    
    if target <= nums[start] {
        return start
    } else if target <= nums[end] {
        return end
    } else {
        return end + 1
    }
}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019年10月03日，如有侵权请联系 cloudcommunity@tencent.com 删除

编程算法