搜索插入位置

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

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

实现思路：

我们使用二分查找来实现，首先初始化两个指针 left 和 right，分别指向数组的第一个元素和最后一个元素的索引，进入循环，当 left <= right 时执行以下操作：计算中间索引 mid = left + (right - left) / 2，如果数组中中间位置的元素 nums[mid] 等于目标值 target，则直接返回 mid，表示找到了目标值，其索引为 mid。如果 nums[mid] < target，说明目标值在数组的后半部分，将 left 更新为 mid + 1，继续在右半部分查找。如果 nums[mid] > target，说明目标值在数组的前半部分，将 right 更新为 mid - 1，继续在左半部分查找。当循环结束时，如果没有找到目标值，此时 left 指向的位置就是目标值应该插入的位置，将其作为结果输出。

具体代码：

import java.util.Scanner;

class SearchInsertPosition {
    public static 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;
            }
        }
        return left;
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int[] nums = new int[4];
        System.out.println("nums=");
        for (int i = 0; i < 4; i++) {
            nums[i] = scanner.nextInt();
        }
        System.out.println("target");
        int target = scanner.nextInt();
        int result = searchInsert(nums, target);
        System.out.println("输出" + result);
        scanner.close();
    }
}

运行截图：