《中英双解》leetCode Search Insert Position(搜索插入位置)

最新推荐文章于 2024-07-28 22:04:38 发布

@我好菜啊

最新推荐文章于 2024-07-28 22:04:38 发布

阅读量135

点赞数

分类专栏： leetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/weixin_46272350/article/details/120864243

版权

leetCode 专栏收录该内容

79 篇文章 0 订阅

订阅专栏

该博客讨论了一种在有序数组中查找目标值并返回其索引或插入位置的问题。文章指出，最佳解决方案是使用二分查找算法，其时间复杂度为O(logn)。博主分享了两种实现方式，包括一个简单的线性搜索和一个优化的二分查找方法。二分查找的关键在于寻找第一个大于等于目标值的元素的索引。博客还提供了详细的代码示例，帮助读者理解如何应用二分查找来解决这个问题。

摘要由CSDN通过智能技术生成

Given a sorted array of distinct integers and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.

You must write an algorithm with O(log n) runtime complexity.

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

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

Example 1:
Input: nums = [1,3,5,6], target = 5
Output: 2
Example 2:
Input: nums = [1,3,5,6], target = 2
Output: 1
Example 3:
Input: nums = [1,3,5,6], target = 7
Output: 4
Example 4:
Input: nums = [1,3,5,6], target = 0
Output: 0
Example 5:
Input: nums = [1], target = 0
Output: 0

这一题偏简单，大家应该都能看懂，我也就不多说了，现在看看我的代码，不过写的不是很好，一会再看看别人的。

class Solution {
    public int searchInsert(int[] nums, int target) {
        int j = 0;
        for(int i = 0;i < nums.length;i++){
            if(nums[i] == target){
                return i;
            } else if(target > nums[i] ){
                j = i + 1;
            }
        }
        return j;
    }
}

这一题最好的解法是二分查找，因为这是有序数组，二分查找可以看看我这一篇博客。

二分查找

思路与算法

假设题意是叫你在排序数组中寻找是否存在一个目标值，那么训练有素的读者肯定立马就能想到利用二分法在 O(logn) 的时间内找到是否存在目标值。但这题还多了个额外的条件，即如果不存在数组中的时候需要返回按顺序插入的位置，那我们还能用二分法么？答案是可以的，我们只需要稍作修改即可。

考虑这个插入的位置 pos，它成立的条件为：
nums[pos−1]<target≤nums[pos]

其中 nums 代表排序数组。由于如果存在这个目标值，我们返回的索引也是pos，因此我们可以将两个条件合并得出最后的目标：「在一个有序数组中找第一个大于等于target 的下标」。

问题转化到这里，直接套用二分法即可，即不断用二分法逼近查找第一个大于等于 target 的下标。下文给出的代码是笔者习惯的二分写法，ans 初值设置为数组长度可以省略边界条件的判断，因为存在一种情况是 target 大于数组中的所有数，此时需要插入到数组长度的位置。

下面看看代码

class Solution{
   public int searchInsert(int[] nums, int target) {
      int n = nums.length;
      int left = 0;
      int right = n - 1;
      int ans = n;
      while(left <= right){
         int mid = ((right - left) >> 1) + left;
         if(target <= nums[mid]){
            ans = mid;
            right = mid - 1;
         } else {
            left = mid + 1;
         }
      }
      return ans;
   }
}