leetcode - 34. Find First and Last Position of Element in Sorted Array

最新推荐文章于 2023-07-13 23:29:53 发布

Zino-齐诺

最新推荐文章于 2023-07-13 23:29:53 发布

阅读量113

点赞数

分类专栏： leetcode 文章标签： leetcode 二分查找

本文链接：https://blog.csdn.net/weixin_41938758/article/details/88911820

版权

leetcode 专栏收录该内容

71 篇文章 1 订阅

订阅专栏

二分查找问题
在这里插入图片描述

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int left(0),right=nums.size()-1,mid,b,p;
        int end=right;
        vector<int> ret(2,-1);
        if(end<0) return ret;
        while(1)
        {
            mid=(left+right)/2;
            if(left>(right-2))//结束条件
            {
                if(nums[left]==target)
                {
                    ret[0]=ret[1]=left;
                    if(nums[right]==target)
                    {
                        ret[1]=right;
                    }
                }else if(nums[right]==target)
                {
                    ret[0]=ret[1]=right;
                }
                return ret;
            }
            if(nums[mid]==target)
            {
                ret[0]=ret[1]=mid;
                b=mid;
                while(1)
                {
                    p=(b+left)/2;
                    if(left>(b-2))
                    {
                        if(nums[left]==target)
                        {
                            ret[0]=left;
                        }else if(nums[b]==target)
                        {
                            ret[0]=b;
                        }
                        break;
                    }
                    if(nums[p]==target&&(!p||nums[p-1]!=target))
                    {
                        ret[0]=p;
                        break;
                    }else if(nums[p]!=target)
                    {
                        left=p;
                    }else{
                        b=p;
                    }
                }
                b=mid;
                while(1)
                {
                    p=(b+right)/2;
                    if(b>(right-2))
                    {
                        if(nums[right]==target)
                        {
                            ret[1]=right;
                        }else if(nums[b]==target)
                        {
                            ret[1]=b;
                        }
                        break;
                    }
                    if(nums[p]==target&&((p==end)||nums[p+1]!=target))
                    {
                        ret[1]=p;
                        break;
                    }else if(nums[p]!=target)
                    {
                        right=p;
                    }else{
                        b=p;
                    }
                }
                return ret;
                
            }else if(nums[mid]<target)
            {
                left=mid;
            }else{
                right=mid;
            }
        }
    }
};