LeetCode 35 搜索插入位置

LeetCode35 搜索插入位置

题目描述

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

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

样例

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

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

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

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

算法分析

• 二分寻找>=target的那一个数
• 最后一个数还小于target，返回nums.size/l++

时间复杂度\(O(log(n))\)

Java代码

class Solution {
    public int searchInsert(int[] nums, int target) {
        int n = nums.length;
        int l = 0;
        int r = n - 1;

        while(l < r){
            int mid = l + r  >> 1;
            if(nums[mid] >= target) r = mid;
            else l = mid + 1;
        }

       
        if(l == n-1 && target > nums[l]) l++;
        return l;
    }
}