leetcode练习题 search-for-a-range

最新推荐文章于 2024-08-20 12:37:20 发布

qq_38303368

最新推荐文章于 2024-08-20 12:37:20 发布

阅读量223

点赞数

分类专栏： leetcode练习文章标签：算法 leetcode

本文链接：https://blog.csdn.net/qq_38303368/article/details/105142331

版权

leetcode练习专栏收录该内容

70 篇文章 0 订阅

订阅专栏

解题思路

二分查找，找到目标后，start向左找左边界，end向右找右边界，若数组中不存在目标元素则返回两个-1.算法复杂度可能不满足O(logn)

代码

class Solution {
public:
    vector<int> searchRange(int A[], int n, int target) {
        int low = 0,high = n - 1;
        int start = -1,end = -1;
        while(low <= high){
            int mid = (low + high) / 2;
            if(A[mid] > target)
                high = mid - 1;
            else if(A[mid] < target)
                low = mid + 1;
            else{
                start = end = mid;
                while(start >= 0 && A[start] == A[mid])
                    start--;
                while(end < n && A[end] == A[mid])
                    end++;
                break;
            }
        }
        vector<int>res;
        if(start == -1 && end == -1){
            res.push_back(-1);
            res.push_back(-1);
        }
        else{
            res.push_back(start + 1);
            res.push_back(end - 1);
        }
        return res;
    }
};