leetcode35. search Insert Position

问题描述:

Given a sorted array 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 may assume no duplicates in the array.

Example 1:

Input: [1,3,5,6], 5
Output: 2

Example 2:

Input: [1,3,5,6], 2
Output: 1

Example 3:

Input: [1,3,5,6], 7
Output: 4

Example 4:

Input: [1,3,5,6], 0
Output: 0

思路:

目的是找到插入的坐标并且返回,如果目标存在则直接返回坐标,如果目标不存在则返回应该插入的坐标(这个时候应该仔细观察一下例子汇总插入的坐标点,为了更准确的答题)

一开始我就想简单啊,遍历一遍就好了啊,复杂度只有O(n)多好

结果我的运行效率巨低无比 _(:з」∠)_

是我太天真:)

二分搜索回忆一下,时间复杂度O(logn)

所以说,看到搜索字样,不妨回忆一下二分搜索。。。

代码1 (线性搜索):

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

代码2:(二分搜索):

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

 看到搜索,要记得二分搜索!!!!

猜你喜欢

转载自www.cnblogs.com/yaoyudadudu/p/9095045.html
今日推荐