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;

    }
};

运行结果：