leetcode

Question34–Search for a Range

給定一個按升序排列的可以有重複元素的數組,同時給出一個目標元素target,返回target在數組中的開始下標和結束下標。若不在數組中,返回[-1,-1]. eg:[1,2,2,4,5,5,7]和5,則返回[4,5]。

算法

  1. 使用二分法,縮小區間,直到中間位置爲target。
  2. 分別在找到target的那個區間的左、右區間進行二分查找,確定target的開始位置和結束位置。

Code

 vector<int> searchRange(vector<int>& nums, int target) {
        int i=0, j=nums.size()-1;
        int start=-1, end=-1;
        while(i<=j){
            int mid=(i+j)/2;
            if(target<nums[mid]) j=mid-1;
            else if(target>nums[mid]) i=mid+1;
            else{
                int p,q;
                p=i;
                q=mid;
                while(p<q){
                    int leftmid=(p+q)/2;
                    if(nums[leftmid]==target) q=leftmid;
                    if(nums[leftmid]<target) p=leftmid+1;
                }
                start=q;

                p=mid;
                q=j;
                while(p<q){
                    int rightmid=(p+q+1)/2;
                    if(nums[rightmid]==target) p=rightmid;
                    if(nums[rightmid]>target) q=rightmid-1;
                }
                end=p;
                break;
            }
        }
        vector<int> range{start,end};
        return range;
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章