LeetCode 35. Search Insert Position 查找插入位置

1.题意

给定有序数组和目标值，如果找到目标值，则返回索引。 如果没有，请返回按顺序插入的索引。

可以假设数组中没有重复项。

题目其实是查找等于或第一个大于target的元素的下标。

有序数组的查找考虑二分查找（Binary Search）。

示例

Example 1:

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

Example 2:

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

2.代码实现

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        if(nums.size() == 0)
            return 0;
        
        int lo = 0;
        int hi = nums.size() - 1;
        
        while(lo <= hi)
        {
            int mi = lo + (hi - lo) / 2 ;
            if(nums[mi] == target)
                return mi;
            else if(nums[mi] < target)
                lo = mi + 1;
            else
                hi = mi - 1;
            
        }
        
        return lo;
    }
};

为什么返回值是lo？

假设二分前原始数组的范围是[首, 末]

在每一次while循环进行前，算法的不变性是：

                                [首，lo)间元素都比target小

                                (hi，末]间元素都比target大

最后一轮循环中，lo = hi

mi = lo = hi

/*
    [首, lo)  [mi]  (hi, 末]
       |       |       |
   比target小  lo   比target大
               hi
*/
if(nums[mi] == target)
    return mi; 
else if(nums[mi] < target) //如果nums[mi] 比 target 小 
    lo = mi + 1; //lo = mi + 1，此时指向 mi + 1, 即(hi, 末]区域的起始位置，比target大的第一个元素
else     //如果nums[mi] 比 target 大 
    hi = mi - 1; //lo不变，恰好指向比target大的第一个元素


// 综上，要返回lo