【Leetcode】之Search Insert Position

最新推荐文章于 2019-05-31 00:39:10 发布

原创最新推荐文章于 2019-05-31 00:39:10 发布 · 362 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

Leetcode 同时被 2 个专栏收录

75 篇文章

订阅专栏

算法练习

72 篇文章

订阅专栏

本文介绍了一个二分查找算法的应用案例，即在一个已排序且不含重复元素的数组中查找目标值的位置，若不存在则返回该值按顺序应当插入的位置。文章提供了一个C++实现示例，并详细解释了算法流程。

一.问题描述

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

二.我的解题思路

拿到这个题目，自然是可以延续上一道题目的思路，使用二分法来尽快找到target。相比于上一题，这道题比较简单。测试通过的程序如下：

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int len=nums.size();
        int idx=-1;
        int left=0;int right=len-1;int mid=0;int left_idx=-1;int right_idx=len; 
        if(target<nums[0]) return 0;
        if(target>nums[len-1]) return len;
        mid=(left+right)/2;

          //首先使用二分法寻找数组nums中可有target,并且记录左右边界：nums[left_idx]<target和nums[right_idx]>target
         while(nums[mid]!=target)
            {
            if(nums[mid]>target)
                    {right=mid;right_idx=right<right_idx?right:right_idx;}
            if(nums[mid]<target)
                   {left=mid;left_idx=left>left_idx?left:left_idx;}
            mid=(left+right)/2;

            if(right-left==1){
                 if(nums[left]==target)
                    {idx = left;break;}
                 if(nums[right]==target)
                    {idx = right;break;}
                    break;

             }
            if(left==right)
                break;
            }
             if(nums[mid]==target)
                idx=mid;
            if(idx==-1)
                return right;
            return idx;
    }
};