[LeetCode] 34、在排序数组中查找元素的第一个和最后一个位置-优快云博客

本文介绍了一种在升序数组中寻找特定元素起始和结束位置的高效算法，采用二分查找策略实现O(logn)的时间复杂度。通过两个辅助函数分别定位目标值的首次和末次出现位置，最终返回这两个位置的索引。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

你的算法时间复杂度必须是 O(log n) 级别。

如果数组中不存在目标值，返回 [-1, -1]。

示例：

输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]

参考代码

class Solution {
public:
    vector<int> searchRange(vector<int> &data, int k) {
        vector<int> res;
        res.push_back(getIndexOfFirstK(data, k));
        res.push_back(getIndexOfLastK(data, k));
        
        return res;
    }
    
    
    int getIndexOfFirstK(vector<int> data ,int k){
        int low = 0;
        int high = data.size() - 1;
        while(low <= high){
            int mid = (low + high) >> 1;
            
            if(data[mid] == k){
                if(mid == 0 || data[mid-1] != k)
                    return mid;
                high = mid - 1;
            }else if(data[mid] < k){
                low = mid + 1;
            }else{
                high = mid - 1;
            }
        }
        
        return -1;
    }
    
    int getIndexOfLastK(vector<int> data ,int k){
        int low = 0;
        int high = data.size() - 1;
        while(low <= high){
            int mid = (low + high) >> 1;
            
            if(data[mid] == k){
                if(mid == data.size()-1 || data[mid+1] != k)
                    return mid;
                low = mid + 1;
            }else if(data[mid] < k){
                low = mid + 1;
            }else{
                high = mid - 1;
            }
        }
        
        return -1;
    }
        
};