您的位置:首页 > 理论基础 > 数据结构算法

leetcode——Search Insert Position

2016-06-15 23:11 363 查看
题目:

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

class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if (nums.empty())
{
return 0;
}
/*找到第一个大于等于target的位置,那么可能需要比较first和first-1的位置
因此,让first从1开始比较合理
*/
if (nums[0] >= target)
{
return 0;
}
//先比较最后一个元素,这样让last从nums.size()开始,防止越界
int first = 1, last = nums.size();
while (first < last)
{
if (nums[first] == target || (nums[first] > target && nums[first - 1] < target))
{
return first;
}

int mid = first + (last - first) / 2;

if (nums[mid] == target || (nums[mid] > target && nums[mid - 1] < target))
{
return mid;
}
else if (nums[mid] < target)
{
first = mid + 1;
}
else
{
last = mid;
}
}

//没有找到大于等于target的元素,那么返回nums.size()
return nums.size();
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息