LeetCode 35. 搜索插入位置

两种方法，第一种直接顺序查找数组中的每一个元素，查找大于或等于目标值的数。时间复杂度O(n)。

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        for(int i=0; i<nums.size(); i++){
            if(nums[i]>=target)
                return i;
        }
        return nums.size();
    }
};

第二种方法是二分查找，对于有序数组，二分法可以将查找的时间复杂度降到O(logn)。二分算法可以自己写，也可以直接用STL中提供的函数lower_bound。

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left=0, right=nums.size()-1;
        int mid=-1;
        while(left<=right){
            mid=(left+right)/2;
            if(nums[mid]==target)
                return mid;
            if(nums[mid]>target)
                right=mid-1;
            else
                left=mid+1;
        }
        return left;
    }
};