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