(力扣) 35. 搜索插入位置

发表于 更新于 分类于 ，

题目描述

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

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

示例 1:

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

示例 2:

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

示例 3:

1
2
输入: nums = [1,3,5,6], target = 7
输出: 4

提示:

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

解题思路

定义左右两个指针 leftleft 和 rightright，分别指向数组的开头和结尾。定义指针 mid=(left+right)/2mid = (left + right)/2，其指向 leftleft 和 rightright 所指向区间中间的数字（向下取整）。如果该数字小于 targettarget，则 targettarget 在该数字右边，将 leftleft 移动到的 midmid 右边；如果该数字大于 targettarget，targettarget 在该数字的左边，将 rightright 移动到 midmid 的左边；如果该数字等于 targettarget，则返回 midmid 的值，即 targettarget 所在的下标。当 leftleft 大于 rightright 时，表示没有在数组中找到 targettarget，于是返回 leftleft 的值，该值表示 targettarget 应该插入的位置。

代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        int mid;
        while(left <= right)
        {
            mid = (left + right)/2;
            if(nums[mid] < target)
            {
                left = mid + 1;
            }
            else if(nums[mid] > target)
            {
                right = mid - 1;
            }
            else
            {
                return mid;
            }
        }
        return left;
    }
};