LeetCode第35题:搜索插入位置

题目描述:

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

示例 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

解题思路:

数组已经有序且无重复元素,很显然应该使用二分查找。
如果target在数组中,最终会返回它的位置。
如果target不在数组中,此时low==high,如果low所指位置<target,则插入位置应为low+1;
否则low即为target应该插入的位置。

C++代码:

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int low = 0,high = nums.size()-1;
        int middle;
        
        //二分查找
        while(low<high){
            middle = (low+high)>>1;
            if(nums[middle]<target)
                low = middle+1;
            else if(nums[middle]>target)
                high = middle-1;
            else
                return middle;
        }
        
        //二分查找未找到target,此时low=high
        //若low位置<target,应前进一格
        //否则low即为插入位置
        if(nums[low]<target)
            return low+1;
        else
            return low;
        
        
    }
};

运行结果:

posted @ 2018-12-30 09:57  朕蹲厕唱忐忑  阅读(120)  评论(0编辑  收藏  举报