Leetcode#35: Search Insert Position

题目描述：

给定一个已排序的数组和一个target值，若数组中有元素的值与target相等，则返回其位序；若找不到，则返回要插入位置的位序。假设数组中无重复元素。

以下是测试用例：

[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6], 0 → 0

思路：这道题的最优解法是采用二分查找法，当然暴力求解也可以，不过复杂度更高，但是思路会更加清楚。

以下给出两种解法，思路步骤都写在注释中，可供参考。若有需要，请访问我的Github获取包含测试程序的C++源码。

//暴力求解法，时间复杂度O(n),空间复杂度O(1)
class Solution 
{
public:
    int searchInsert(vector<int>& nums, int target) 
    {
        //插入到第一个位置：若数组为空，直接返回0位置
        if(nums.empty())
        {
            return 0;
        }
        //插入到最后一个位置：若数组的最后一个元素都比target小，直接返回数组的长度
        else if(nums[nums.size() - 1] < target)
        {
            return nums.size();
        }

        for(int i = 0; i < nums.size(); i++)
        {
            //正好找到的情况：若数组中序号为i的元素和target相等，则返回i
            if(nums[i] == target)
            {
                return i;
                break;
            }
            //要插入到中间的情况：若序号为i的元素比target大，则插入到当前i的位置，跳出循环
            else if(nums[i] > target)
            {
                return  i;
                break;
            }
        }
     }
};

//二分法求解，时间复杂度O(logn),空间复杂度O(1)，最优解法
class Solution 
{
public:
    int searchInsert(vector<int>& nums, int target) 
    {
        //插入到第一个位置：若数组为空，直接返回0位置
        if(nums.empty())
        {
            return 0;
        }
        int left = 0;
        int right = nums.size() - 1;
        while(left <= right)
        {
            int mid = (right + left)/2;

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

            }
        }
        //返回的正好是插入的位序
        return left;
    }
};