搜索插入位置
思路:直接套用之前的二分模板,二分有两套模板。
代码:
调边界问题有点破防了,直接
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
return lower_bound(nums.begin(),nums.end(),target)-nums.begin();//因为 lower_bound 返回的是一个迭代器,它要减去nums.begin()这个迭代器得到的差值才是下标的编号。
}
};
回顾一下lower_bound:
last:注意是最后一个元素之后的迭代器!
第二天回来好好审视这个题。
正常写二分就是
但这个题目,还多了个额外的条件,即如果不存在数组中的时候需要返回按顺序插入的位置,那我们还能用二分法么?答案是可以的,我们只需要稍作修改即可。
思考了一下,其实修改也可以不用,看代码吧
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int n=nums.size();
int l=0,r=n;
while(l<r)
{
int mid = (l+r)>>1;
if(nums[mid]<target) l=mid+1;
else
{
r=mid;
}
}
return r;
}
};
设初值的时候设置一下r在数组的最大值右边就好,可以处理边界情况。