[LeetCode] #35 搜索插入位置

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

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

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

输出: 2

重点在于O(log n),所以需要在一个循环解决问题

那么,我们逐个比较,当target等于数组元素时返回当前元素下标;小于时需要插入到当前元素前面,也就是当前元素的位置,所以也返回当前元素下标。

如果循环结束仍未返回,则说明target大于所有元素,需要插入到最后,所以返回数组的长度。

class Solution {
    public int searchInsert(int[] nums, int target) {
        for(int i = 0; i < nums.length; i++){
            if(nums[i] >= target) return i;
        }
        return nums.length;
    }
}

当然,对于有序序列的查找首选还是二分法

class Solution {
    public int searchInsert(int[] nums, int target) {
        int low = 0;
        int high = nums.length - 1;
        while (low <= high) { 
            int mid = low + (high - low) / 2; 
            if (nums[mid] > target) high = mid - 1; 
            else if (nums[mid] < target) low = mid + 1; 
            else return mid;
        }
        return high + 1;
    }
}

知识点:

总结:

posted @ 2021-07-24 12:52  1243741754  阅读(22)  评论(0编辑  收藏  举报