【11.搜索插入位置】

49 阅读1分钟

题目

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

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

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1

题解

方式一:二分(变式)

复杂度:O(logn)

// 其实就是找到第一个大于等于target的下标
public int searchInsert(int[] nums, int target) {
    int len = nums.length;
    int l = 0;
    int r = len - 1;
    int ans = len; // 不用考虑越界问题
    while (l <= r) {
        int mid = l + (r - l) / 2;
        if (target <= nums[mid]) {
            ans = mid;
            r = mid - 1;
        } else {
            l = mid + 1;
        }
    }
    return ans;
}

总结

算法:二分法二分变式