Go语言题解LeetCode35搜索插入位置示例详解

题目描述

原题链接 :

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;
   }
}

代码

int searchInsert(int* nums, int numsSize, int target){
   if(nums[0]>=target)
   {
       return 0;
   }
   else if(nums[numsSize-1]<target)
   {
        return numsSize;
   }
   else if(nums[numsSize-1]==target)
   {
       return numsSize-1;
   }
   else{
        int r=numsSize-1;
        int i=0;
   while(i+1!=r)
   {
       int mid=(i+r)/2;
   if(nums[mid]>target )
{
   r=mid;
}
       else if(nums[mid]<target)
                i=mid;
       else{
                return mid;
}
   }
           return i+1;

}
}

来源：https://juejin.cn/post/7173274268069953572