leet code 35.搜索插入位置

原创

隙间光点 2023-09-10 10:48:42 博主文章分类：算法 ©著作权

文章标签 数组边界条件数组元素 文章分类 代码人生

©著作权归作者所有：来自51CTO博客作者隙间光点的原创作品，请联系作者获取转载授权，否则将追究法律责任

简单题蕴含大学问

link leet code 35.搜索插入位置

思路历程

学习算法已有时日，再做这一道简单程度的二分题目时-发现还是不能游刃有余地掌握。题目中要求：需要在数组中找到目标值，并返回其索引，如果目标值不存在于数组中的话，返回其将会被顺序插入的位置。那么有两种情况

目标值在数组中存在
目标值在数组中不存在

如何兼顾这两种情况呢？

首先考虑目标值在数组中存在的情况。

int mid = (low + high) / 2;
if (arr[mid] == target) {
    // 找到目标值，退出循环
} else if (arr[mid] < target) {
    low = mid + 1; // 更新low
} else {
    high = mid - 1; // 更新high
}

明确目标：也就是说，需要找到目标值所在的索引位置。如果不在的话，返回下一位。
代码如何实现？

class Solution {
    public int searchInsert(int[] nums, int target) {
        int len = nums.length;
        // 初始化结果
        int ans = len;
        int left = 0,right = len - 1;

        while(left <= right) {
            int mid = left + (right - left) / 2;
            if(nums[mid] >= target) {
                // 这里用 ans 记录查找的目标元素的位置。
                // 一定不会遗漏
                // 当数组元素存在的时候，ans 输出目标值所在位置的索引
                // 当数组元素不存在的时候，ans 数组最接近目标值所在位置的索引
                // 初始化 ans = len，即当目标值大于数组中元素的最大值的时候，需要插入的位置是  len.
                ans = mid;
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }

        return ans;
    }
}