LeetCode 35. Search Insert Position (搜索插入位置)

LeetCode 35. Search Insert Position (搜索插入位置)

题目

链接

https://leetcode-cn.com/problems/search-insert-position/

问题描述

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

你可以假设数组中无重复元素。

示例

输入: [1,3,5,6], 5
输出: 2

思路

有序数组，找一个位置，那么第一想法就是遍历，只需要on，之后考虑到有序，二分的效率会更高，设置left和right两个指针直接二分即可。

注意的地方两点：

1. 初始中mid的计算，采用 r-l 来避免溢出，采用右移确保效率。
2. 应该采用 l<=r 的判定条件

复杂度分析

时间复杂度 O(logn)
空间复杂度 O(1)

代码

Java

    public int searchInsert(int[] nums, int target) {
        if (nums.length == 0) {
            return 0;
        }
        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 if (nums[mid] > target) {
                right = mid - 1;
            }
        }
        return left;
    }