LintCode_060_搜索插入位置

问题描述

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

你可以假设在数组中无重复元素。

样例

[1,3,5,6]，5 → 2

[1,3,5,6]，2 → 1

[1,3,5,6]， 7 → 4

[1,3,5,6]，0 → 0

求解

经典的“二分查找”在未找到匹配值时，一般返回 -1，这里在经典算法上稍加改造便能符合题目需求

class Solution
{
  public:
    int searchInsert(vector<int> &A, int target)
    {
        int left = 0;
        int right = A.size() - 1;

        while (left <= right)
        {
            int mid = left + (right - left) / 2;
            if (target < A[mid])
            {
                right = mid - 1;
            }
            else if (A[mid] < target)
            {
                left = mid + 1;
            }
            else
            {
                return mid;
            }
        }

        return left;
    }
};

我们还有STL这个“外挂”，只需一行代码就能解决问题

class Solution2
{
  public:
    int searchInsert(vector<int> &A, int target)
    {
        return lower_bound(A.begin(), A.end(), target) - A.begin();
    }
};