LeetCode:数组之搜索插入位置

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

你可以假设数组中无重复元素。

示例 1:

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

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

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

输入: [1,3,5,6], 0
输出: 0

解法:

(1)遍历搜索法

这种方法简单粗暴,直接进行全局遍历寻找目标元素

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

(2)二分查找法

对于题目中这种有序序列,当然可以使用经典的二分查找算法

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

需要注意的是,这里最后只能返回left,返回right会报错。

发布了80 篇原创文章 · 获赞 101 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/BigDream123/article/details/103225777