LeetCode:34. Find First and Last Position of Element in Sorted Array(在排序数组中查找元素的第一个和最后一个位置)

题目

原题链接

算法

• 题目是对有序数组进行下标的查找，而且要求时间复杂度为$log(n)$,因而比较容易想到二分法
• 以往的二分法是查找到一个元素下标就得到题目的结果，但题目要求返回最小目标元素下标和最大目标元素下标，因而在二分法查找结束后还要进行下标指针的移动
• 算法过程大致如下：

AC代码（c++ ）

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int len = nums.size();
        vector<int> result;
        if(len==0)
        {
            result.push_back(-1);
            result.push_back(-1);
            return result;
        } 
        if(len==1&&nums[0]==target)
        {
            result.push_back(0);
            result.push_back(0);
            return result;
        } 
        int i=0,j=len-1,m;
        while(i<=j)
        {
            m=(i+j)/2;
            if(nums[m]==target)
                break;
            if(nums[m]<target)
                i=m+1;
            else if(nums[m]>target)
                j=m-1;
        }
        cout<<i<<j;
        if(i>j)
        {
            result.push_back(-1);
            result.push_back(-1);
            return result;
        } 
        while(m>=0&&nums[m]==target)
            m--;
        result.push_back(m+1);
        m=m+1;
        while(m<len&&nums[m]==target)
            m++;
        result.push_back(m-1);
        return result;
    }
};