leetcode 之 35. 搜索插入位置

151 阅读1分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第5天,点击查看活动详情

题目描述

原题链接 :

35. 搜索插入位置 - 力扣(LeetCode) (leetcode-cn.com)

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

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

 

提示:

  • 1 <= nums.length <= 10^4
  • -10^4 <= nums[i] <= 10^4
  • nums 为 无重复元素 的 升序 排列数组
  • -10^4 <= target <= 10^4

思路分析

首先明确题目的含义:(1)数组是有序的;(2)如果含有与目标值相等,则返回目标值下标,若不同,则按顺序排序获取下标

思路:采用二分搜索法的策略,获取中间数据的大小,缩短数组的大小定位区间,如在数组中间的前一段,数组中间的后一段

获取下标i的值arrs[i]>=target,进行相应的下标返回

AC 代码

class Solution {
    public int searchInsert(int[] nums, int target) {
        int index = (nums.length / 2);
        if (nums[index] >= target) {
            return compareByIndex(nums, 0, index+1, target);
        } else
            return compareByIndex(nums, index+1, nums.length, target);

    }

    private int compareByIndex(int[] nums, int start, int end, int target) {
        for (int i = start; i < end; i++) {
            if (nums[i] >= target) {
                return i;
            }
        }
        return end;
    }
}

四、总结

这种查找位置的,肯定二分法是合适的,即使不能直接套用二分法的公式,也是二分法的思想,变种。

参考

❤️思维导图整理: 总结了二分查找的通用模板写法, 彻底解决几个易混淆问题❤️ - 搜索插入位置 - 力扣(LeetCode) (leetcode-cn.com)

优先考虑边界情况,红蓝标记解法,简单易懂 - 搜索插入位置 - 力扣(LeetCode) (leetcode-cn.com)