力扣算法题-搜索插入位置

35. 搜索插入位置

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

请必须使用时间复杂度为 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

为了在给定的排序数组中找到目标值或确定其插入位置，并满足时间复杂度为O(log n)，我们可以使用二分查找算法。二分查找算法通过不断将数组分成两半来查找目标值，每次比较中间元素与目标值的大小，从而决定是继续在左半部分查找还是在右半部分查找。

如果目标值不存在于数组中，二分查找结束时，我们将知道目标值应该插入的位置，即最后一次比较时中间元素的索引（如果目标值小于中间元素）或中间元素的下一个索引（如果目标值大于中间元素且中间元素是数组的最后一个元素，或者中间元素之后还有元素且目标值小于这些元素）。

以下是使用Java实现的代码：

public class Solution {  

    public int searchInsert(int[] nums, int target) {  

        int left = 0;  

        int right = nums.length - 1;  

         

        while (left <= right) {  

            int mid = left + (right - left) / 2; // 防止溢出  

            if (nums[mid] == target) {  

                return mid; // 找到目标值，返回索引  

            } else if (nums[mid] < target) {  

                left = mid + 1; // 目标值在右半部分  

            } else {  

                right = mid - 1; // 目标值在左半部分或不存在（且小于当前中间值）  

            }  

        }  

         

        // 循环结束时，left 的值就是目标值应该插入的位置  

        return left;  

    }  

    public static void main(String[] args) {  

        Solution solution = new Solution();  

        int[] nums = {1, 3, 5, 6};  

        int target = 5;  

        System.out.println(solution.searchInsert(nums, target)); // 输出: 2  

        target = 2;  

        System.out.println(solution.searchInsert(nums, target)); // 输出: 1  

        target = 7;  

        System.out.println(solution.searchInsert(nums, target)); // 输出: 4  

    }  

}