LeetCode-数组-35. 搜索插入位置

描述

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

请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2
示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1
示例 3:

输入: nums = [1,3,5,6], target = 7
输出: 4
示例 4:

输入: nums = [1,3,5,6], target = 0
输出: 0
示例 5:

输入: nums = [1], target = 0
输出: 0

提示:

1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 为无重复元素的升序排列数组
-104 <= target <= 104

思路一：暴力解法

没有达到题目要求时间复杂度O(logn) 的要求
时间复杂度为：O(n)
空间复杂度为：O(1)

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int res=0;  //用来返回的结果
        for(int i=0;i<nums.size();i++){
            if(nums[i]==target){
                return i;
            }else if (nums[i]>target&&i!=nums.size()-1){
                if(i<nums.size()-1){
                    return i;
                }else{
                    return i+1;
                }
            }else if (i==nums.size()-1){
                if(nums[i]>target){
                    return i;
                }else{
                     return i+1;
                }
            }
        }
        return 0;
    }
};

思路二：二分查找法

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int res=0;  //用来返回的结果
        int left = 0;
        int right = nums.size()-1;  //定义target在[left,right]中
        while(left<=right){
            int mid = left +((right-left)/2);   //防止内存溢出，等价(left+right)/2
            if(nums[mid]<target){           //目标值在右区间中
                left = mid + 1;
            }else if (nums[mid]>target){    //目标在左区间中
                right = mid - 1; 
            }else if (nums[mid]==target){   //当找到目标值
                return mid;
            } 
        }
        //分别处理如下四种情况
        // 目标值在数组所有元素之前  [0, -1]
        // 目标值等于数组中某一个元素  return middle;
        // 目标值插入数组中的位置 [left, right]，return  right + 1
        // 目标值在数组所有元素之后的情况 [left, right]， return right + 1
        return right+1;
    }
};