leetcode 35. 搜索插入位置

描述

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

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

示例 1:

输入: [1,3,5,6], 5
输出: 2
示例 2:
输入: [1,3,5,6], 2
输出: 1
示例 3:
输入: [1,3,5,6], 7
输出: 4
示例 4:
输入: [1,3,5,6], 0
输出: 0

分析

蛮简单的一道题，很容易想到了二分查找，找到了就找到了，没找到分三种情况：
一种就是在两个数之间，就是失配的最后i==j却不等于target的情况；
另一种是比最小的还小，位置在1；
还有就是比最大的大，位置在最后
其实这题有bug，应该说明已经是顺序的数组了，否则还要先排序才能用二分，这里没排序也是AC的

代码

int searchInsert(vector<int>& nums, int target) {
	int i = 0, j = nums.size()-1;
	int index = -1;
	while (i <= j)
	{
		int mid = (i + j) / 2;
		if (nums[mid] == target) { index = mid; break; }
		else if (nums[mid] < target) { i = mid + 1; }
		else j = mid - 1;
	}
	if (index == -1)
	{
		if (i == 0 && nums[i] > target)index = 0;
		else if (i == nums.size() - 1 && nums[i] < target)index = nums.size();
		else index = i;
	}
	return index;
}