Leetcode——35 搜索插入位置

Leetcode——35 搜索插入位置

题目描述

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

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

题解

很显然的二分查找，不过有一个小细节，题目相对于Leetcode704的改变为当不存在的时候输出它按序应该插入的位置。我的思路是尽可能查找从左靠近target这个数的位置，从左靠近的意思是比如target是6，序列是[1,2,3,4,5,7,8]，那么从左靠近的下标就是4（即5这个数）。为什么这么做呢？实际上插入的位置（当target不存在时）就是这个尽可能从左靠近的target的数的下标+1。

那么考虑边界情况

1. 需要在最后一数插入，比如[1,2,3,4,5]而target是6，那么二分结束后l一定是4（即5），直接加一即可。
2. 需要在第一个数就插入，即target小于序列中所有的数，二分结束后l是0，加一答案就不对了，因此这个情况只需要在开头特判即可。

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        if(target < nums[0]) return 0;
        int l = 0;
        int r = nums.size() - 1;
        while(l < r) {
            int mid = l + r + 1 >> 1;
            if (nums[mid] <= target) l = mid;
            else r = mid - 1;
        }
        if(nums[l] == target) return l;
        else return l + 1;
    }
};