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};
}
};