Binary Search:35. Search Insert Position

来源:互联网 发布:矩阵的迹是什么 编辑:程序博客网 时间:2024/06/06 03:38

给出一个数组,和一个target,如果target在数组中,返回该数字的下标,如果不存在在数组中,返回插入这个数字的下标。

class Solution {public:    int searchInsert(vector<int>& nums, int target) {        if(nums[0] >= target)    return 0;        if(nums[nums.size()-1] < target)    return nums.size();        for(int i = 1; i < nums.size(); ++i)        {            if(nums[i] == target || nums[i] > target)   return i;        }    }};
二分搜索的方法:

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


原创粉丝点击