在有序数组中搜索插入位置

最新推荐文章于 2023-01-22 22:45:21 发布

原创最新推荐文章于 2023-01-22 22:45:21 发布 · 592 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二分法 #c++

C++ 专栏收录该内容

2 篇文章

订阅专栏

本文介绍了如何在已排序的数组中找到目标值的索引或其应插入的位置。提供了两种方法，其中方法二是利用二分查找算法进行高效查找。

题目描述：
给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。 假设数组中无重复元素。
来源

方法一：最好理解的方法

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

方法二：二分查找法

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