35. Search Insert Position

题目描述

Given a sorted array of distinct integers 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 must write an algorithm with $O(logn)$ runtime complexity.

---

思路

首先，我们注意到该数组已排序，且题目对时间复杂度有要求，为$O(logn)$。显然，我们需要用二分查找。但是，需要注意的是，若没有找到目标值，我们要返回目标值待插入的下标。因此，我们需要对二分查找进行一些修改。

---

二分查找

我们首先来看二分查找。
一个升序数组nums，我们要查找target的值为65。

我们声明三个变量left，right和mid。

• left指target所在区间的最左端元素的数组下标
• right指target所在区间的最右端元素的数组下标
• mid指target所在区间的中间元素的数组下标
  $$mid=\frac{left+right}{2}$$

显然，nums[mid]将数组nums分成左和右两个区间。接着，我们将nums[mid]与target比较大小，

• 若nums[mid] > target，则target位于左区间内，调整right为mid - 1。
• 若nums[mid] < target，则target位于右区间内，调整left为mid + 1。
• 若nums[mid] = target，则恰好找到target对应的数组下标，返回mid。

然后，我们重复以上内容，不断缩小区间，直至找到target（若存在）。

---

因为数组nums按升序排列，所以，如果target在数组中的话，一定在数组首、末元素组成的区间内。因此，我们给left赋值0（第一个数组元素的下标），给right赋值nums_size - 1（sums_size为数组元素个数。right的值意味着最后一个数组元素的下标，此处rihgt的值为9）。此时，mid的值为4。

• left = 0
• right = 9
• mid = 4

$$\begin{array}{rl}& \because nums[4]=58,target=65\\ & \therefore nums[4]<target\\ & \therefore left=mid+1=5\end{array}$$

---

• left = 5
• right = 9
• mid = 7

$$\begin{array}{rl}& \because nums[7]=73,target=65\\ & \therefore nums[7]>target\\ & \therefore right=mid-1=6\end{array}$$

---

• left = 5
• right = 6
• mid = 5

$$\begin{array}{rl}& \because nums[5]=65,target=65\\ & \therefore nums[5]=target\\ & \therefore return5\end{array}$$

---

理解了二分查找的思想，我们就能完成代码了。

/* 
BinarySearch()实现二分查找
参数：nums -- 数组地址；nums_size -- 数组长度；target -- 目标值
返回：若能找到target，返回对于的下标；若不能，返回-1。
*/
int BinarySearch(int* nums, int nums_size, int target)
{
	int left, right, mid;
	
	left = 0;                               // 第一个数组元素的下标
	right = nums_size - 1;                  // 最后一个数组元素的下标
	while (left <= right)
	{
		mid = left + (right - left) / 2;    // 防止溢出int型
		if (nums[mid] > target)             // target位于左区间
			right = mid - 1;
		else if (nums[mid] < target)        // target位于右区间
			left = mid + 1;
		else if (nums[mid] == target)       // 找到target
			return mid;
	}

	return -1;
}

值得注意的是，我们将mid的表达式做了一些调整。
原来：
$$mid=\frac{left+right}{2}$$
现在：
$$mid=left+\frac{right-left}{2}$$
原因：left，right和mid都是int型，它们之间的数学运算结果也是int型。假如数组很大，那么left和right的值也可能会很大。因此，$left+right$的结果可能会超出int的最大值，发生溢出。
考虑到溢出的风险，我们修改了表达式，数学运算的结果依然不变。同时$right-left$的结果必然小于int的最大值，避免了溢出的风险。

---

解决题目

现在，我们回到这道题目上来。如果说target在nums中存在，那么与二分查找一样。但是，如果不存在，题目要求我们返回插入位置的下标。
我们将部分代码进行修改：
原代码：

if (nums[mid] > target)             
	right = mid - 1;
else if (nums[mid] < target)        
	left = mid + 1;
else if (nums[mid] == target)
	return mid;

修改后：

if (nums[mid] >= target)
	right = mid - 1;
else
	left = mid + 1;

---

修改后，无论nums中是否存在target，程序都会运行到$left+1=right$这一步，即left与right相邻。

• target存在
  仍以上面二分查找的题目为例，target仍为65。
  
  程序继续运行，left最终会来到65。此时，left的值即为所求。
• target不存在
  将target改为66。
  
  程序继续运行，left的值要加一，即target插入位置的下标。此时，left的值也为所求。

总结：无论target是否在数组中，最终left的值都能满足要求。所以，我们最终返回left。

---

完整代码

int searchInsert(int* nums, int numsSize, int target){
    int l, r, mid;

    l = 0; 
    r = numsSize - 1;
    while (l <= r)
    {
        mid = l + (r - l) / 2;
        if (nums[mid] >= target)
            r = mid - 1;
        else
            l = mid + 1;
    }

    return l;
}

---

（完）