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

思路

可以用二分查找while(low <= high),循环结束时low的值即为插入位置。

• 时间复杂度:O(log₂n)
• 空间复杂度:O(1)

代码

public int searchInsert(int[] nums, int target) { // 总时间复杂度O(log₂n)
    if (nums.length == 0) {
        return 0;
    }
    int low = 0;
    int high = nums.length - 1;
    while (low <= high) { // 时间复杂度O(log₂n)
        int mid = low + (high - low) / 2;
        if (nums[mid] < target) {
            low = mid + 1;
        } else if (nums[mid] > target) {
            high = mid - 1;
        } else {
            return mid;
        }
    }
    // 循环结束时low的值即为插入位置
    return low;
}