LeetCode:Search Insert Position

最新推荐文章于 2021-05-03 16:49:38 发布

yao_hust_iAC

最新推荐文章于 2021-05-03 16:49:38 发布

阅读量560

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： algorithm leetcode

本文链接：https://blog.youkuaiyun.com/yao_wust/article/details/40537091

LeetCode 专栏收录该内容

89 篇文章

订阅专栏

本文介绍了一种基于二分查找的算法实现，该算法用于在一个已排序的数组中寻找目标值的位置，若未找到则返回目标值应插入的位置。通过几个实例展示了算法的工作原理及应用场景。

问题描述：

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.

Here are few examples.
[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6], 0 → 0

解题思路:二分查找。如果没有找到，比较A[medium]与target的大小。如果A[medium]大于target，则返回medium+1。如果小于，则返回medium。

代码：

int head = 0;
    int tail = n-1;
    int medium;
    while(head <= tail)
    {
        medium = (head + tail) / 2;
        if(A[medium] == target)
            return medium;
        /*if(tail - head == 1 && A[tail] != target && A[head] != target)
        {
            if(target > A[tail])
                return tail + 1;
            else if(target > A[head] && target < A[tail])
                return tail;
            else
                return head;
        }*/
        if(A[medium] > target)
            tail = medium - 1;
        if(A[medium] < target)
            head = medium + 1;
    }
    if(head == tail)
    {
        if(target > A[tail])
            return tail+1;
        else
            return head;
    }