leetcode笔记：Search Insert Position

Posted 2020-09-30 ljbguanli

一. 题目描写叙述

Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.

You may assume no duplicates in the array.

Here are few examples.

[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应该插入到该数组中的位置下标。这个过程中。假设该数组中没有反复元素。

和Search for a Range 的解法相似。二分查找就可以。可能出问题的地方是对边界条件的处理。

三. 演示样例代码

#include <iostream>
#include <vector>

using namespace std;

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int n = nums.size(), low = 0, high = n - 1, midIndex = 0;
        while (low <= high)
        {
            midIndex = (low + high) / 2;
            if (nums[midIndex] == target)
                return midIndex;
            else if (nums[midIndex] > target)
                high = midIndex - 1;
            else
                low = midIndex + 1;
        }
        // 边界
        if (low > n)
            return n;
        if (high < 0)
            return 0;
    }
};

解题结果：

四. 小结

二分查找的经典题目，对边界的情况须要额外考虑。该题实际上是实现：std::lower_bound() 的功能。