35. 搜索插入位置

最新推荐文章于 2024-10-20 17:09:42 发布

胡矣

最新推荐文章于 2024-10-20 17:09:42 发布

阅读量77

点赞数 1

分类专栏： LeetCode 算法文章标签：数组 java 二分查找 leetcode题解力扣题解

本文链接：https://blog.csdn.net/yirentianran/article/details/122112442

版权

算法同时被 2 个专栏收录

142 篇文章 0 订阅

订阅专栏

LeetCode

130 篇文章 0 订阅

订阅专栏

题目

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2

解题思路

二分查找

一个标准的二分查找，只是要求未查找到target时返回插入的位置，即left的位置
nums[mid] == target,找到target
nums[mid] < target,需向右查找
nums[mid] > target,需向左查找

代码

class Solution {
  public int searchInsert(int[] nums, int target) {
    int l = 0, r = nums.length - 1;
    while (l <= r) {
      // 计算中间元素位置
      int mid = l + ((r - l) >> 1);
      // 找到target直接返回
      if (nums[mid] == target) return mid;
      // 如果nums[mid]<target说明target应该在mid右侧
      else if (nums[mid] < target) l = mid + 1;
      // 如果nums[mid]>target说明target应该在mid左侧
      else r = mid - 1;
    }
    return l;
  }
}