leetcode-34. 在排序数组中查找元素的第一个和最后一个位置(改进的思路)

34. 在排序数组中查找元素的第一个和最后一个位置

第一次写这道题的时候时候耗时特别长,本次的改进也是在上一次的基础上改进。具体可见超时版本
这次的改动也是特别小的,在target=nums[mid]时,第一次是把更新left和right的操作放在了一起,其实letf只用使用左区间更新,right只需要使用右区间更新就可以了。放在一起之后好多层的递归是完全没有必要的,所以通过设置flag来区别是在左区间还是右区间里查找更新。

class Solution {
public:
    int left=0x3f3f3f3f,right=0;//记录最终答案的区间
    void partition(int low,int high,vector<int>nums,int target,int flag)
    {// 就是把while循环改成了递归的样子,就不用既有循环又有递归那样复杂
        if(low>high)return;//函数出口
        int mid=(high-low)/2+low;// 同样的是防止溢出
        if(target>nums[mid])partition(mid+1,high,nums,target,flag);//在右区间查找
        else if(target<nums[mid])partition(low,mid-1,nums,target,flag);//在左区间查找
        else// 与简单的二分查找的不同之处
        {
            if(flag==0)// 只在左区间查找,并更新left
            {
                 left=mid;
                partition(low,mid-1,nums,target,flag);
            }
            else 
            {
                right=mid;//更新right
                partition(mid+1,high,nums,target,flag);
            }
        }
    }
    vector<int> searchRange(vector<int>& nums, int target) 
    {
        partition(0,nums.size()-1,nums,target,0);
        partition(0,nums.size()-1,nums,target,1);
        if(left==0x3f3f3f3f)return vector<int>(2,-1);
        else return vector<int>{left,right};
    }
};
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值