力扣_数组2—搜索插入的位置

题目

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

关键点

• 二分查找（题目要求复杂度为O(log n)）
• 二分法复杂度分析：
  1. 对于顺序查找：每次查找完毕下次查找时数组长度相对于上一次减1
  2. 对于二分查找：每次查找完毕下次查找时数组长度相对于上一次减半
  完成第k次查找后，数组长度为n/2^k。考虑最坏情况，二分查找找到剩下一个元素，则次数数组长度为1=n/2^k，即k=log_2 n。
• 对于本题，要求若没有找到需要返回插入的位置。此时可以这么考虑：
  1. 若目标元素不存在，则一直会找到待查找数组长度为变为1，假设此时左右边界为lb，rb（lb = rb）。
  2. 若这个数大于目标元素，则这个数的前一个数一定小于目标元素，此时lb即为插入的位置
  3. 若这个数小于目标元素，则这个数的后一个数一定大于目标元素，此时lb+1即为插入的位置

代码

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        // int n = nums.size();
        // for(int i = 0; i < n; i++){
        //     if (nums[i] < target){
        //         continue;
        //     }
        //     else
        //         return i;
        // }
        // return n;

        // 二分法
        int n = nums.size();
        int l = n;
        int lb = 0, rb = n-1;
        int p;
        while(rb >= lb){
            p = lb+(rb-lb+1)/2;
            if(nums[p] == target){
                return p;
            }
            else if (nums[p] > target){
                rb = p-1;
            }
            else{
                lb = p+1;
            }
        }
        return lb;
    }
};