leetcode 搜索插入位置

题目

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。

如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

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

示例 1:

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

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

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

解题思路

从题目要求的时间复杂度，我们可以看出这一道二分查找的题目，传统的二分查找一般是搜寻给定数组在数组的位置，这道题是二分的变形，所给的数组可能不在数组里，这时候需要给他们找到在有序数组位置,即在一个有序数组中找第一个大于等于target 的下标。

代码

 public int searchInsert(int[] nums, int target) {
        // 第一个大于等于target的位置
        int id = binarySearch(nums, target);
        return id;
    }
    // 二分查找traget在nums中的位置
    public int binarySearch(int[] nums, int target) {
        int left = 0, right = nums.length - 1;
        // 找不到，设置为最后一个位置 + 1
        int ans = nums.length;
        while (left <= right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] >= target) {
                right = mid - 1;
                ans = mid;
            } else {
                left = mid + 1;
            }
        }
        return ans;
    }

总结

一般看到数组有序，或者阶段有序，首先想到二分，注意二分查找结束的条件，以及如何折半查找。