LeetCode-剑指 Offer II 068. 查找插入位置

83 阅读1分钟

1. 描述

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

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

2. 示例

示例 1

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

示例 2

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

3. 答案

class Solution {
    public int searchInsert(int[] nums, int target) {
        if(nums[0] >= target) {
            return 0;
        }
        if(nums[nums.length - 1] == target) {
            return nums.length - 1;
        }
        if(nums[nums.length - 1] < target) {
            return nums.length;
        }
        int left = 1;
        int right = nums.length - 1;
        while(left < right) {
            int mid = (left + right) / 2;
            if(nums[mid] == target) {
                return mid;
            } else if(nums[mid] > target) {
                if(nums[mid - 1] < target) {
                    return mid;
                }
                right = mid;
            } else {
                if(nums[mid + 1] > target) {
                    return mid + 1;
                }
                left = mid + 1;
            }
        }
        return left;
    }
}
  • 标签:二分查找
  • 时间复杂度:O(logn)
  • 空间复杂度:O(1)

LeetCode:剑指 Offer II 068. 查找插入位置