Algorithm之路三十五：Search Insert Position

最新推荐文章于 2022-04-24 22:29:29 发布

原创最新推荐文章于 2022-04-24 22:29:29 发布 · 109 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种利用二分查找法解决在有序数组中搜索目标元素的问题，并提供了一个C++实现示例。当目标元素不存在于数组中时，该算法能够确定其正确的插入位置。

题目：

给出一个从小到大排号序的数组nums，给出一个int型的target，要求如果target在nums中，则返回target的索引，如果不在，则返回target应该插入的位置。

举例：（例子来源于https://leetcode.com/problems/search-insert-position/description/）

Example 1:

Input: [1,3,5,6], 5
Output: 2

Example 2:

Input: [1,3,5,6], 2
Output: 1

Example 3:

Input: [1,3,5,6], 7
Output: 4

Example 1:

Input: [1,3,5,6], 0
Output: 0

思路：

二分法

代码：

#include <iostream>
#include <cstdlib>
#include <vector>
using namespace std;
int searchInsert(vector<int>& nums, int target)
{
	int length = nums.size();
	if(length == 0) return 0;
	if(target < nums[0]) return 0;
	if(target > nums[length - 1]) return length;
	int head = 0,tail = length,center;
	while(head < tail)
	{
		center = (head + tail)/2;
		if(nums[center] == target) return center;
		if(nums[center] < target) head = center;
		else tail = center;
		if(head + 1 == tail)
		{
			if(nums[head] == target) return head;
			else return tail;
		}
	}
}
int main()
{
	vector<int> nums;
	nums.push_back(1);
	nums.push_back(3);
	nums.push_back(5);
	nums.push_back(6);
	cout<<searchInsert(nums, 0)<<endl;
	return 0;
}