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

最新推荐文章于 2024-11-08 13:05:39 发布

原创最新推荐文章于 2024-11-08 13:05:39 发布 · 117 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

45 篇文章

订阅专栏

本文详细解析了LeetCode第34题“在有序数组中查找元素的第一个和最后一个位置”，介绍了如何使用二分查找算法在O(logn)的时间复杂度内解决此问题。通过示例说明了算法的具体实现过程，包括定位目标元素并确定其范围。

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

Given an array of integers nums sorted in ascending order, find the starting and ending position of a given target value.Your algorithm’s runtime complexity must be in the order of O(log n).If the target is not found in the array, return [-1, -1].

Example 1:
Input: nums = [5,7,7,8,8,10], target = 8
Output: [3,4]
Example 2:
Input: nums = [5,7,7,8,8,10], target = 6
Output: [-1,-1]

题目大意：
给定一个有序数组，返回要寻找的值的起始和终止位置（可能有多个），若不存在则返回两个-1.

解题思路：
O(logn)且数组有序可以自然想到二分查找，找到指定元素后就从该位置向两边扩展直到不相等为止。

代码：

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int l = 0, r = nums.size() - 1;
        vector<int> res;
        while(l <= r)
        {
            int mid = l + (r - l) / 2;
            if(nums[mid] == target)
            {
                l = mid, r = mid;
                while(l > 0 && nums[l - 1] == target)
                    --l;
                while(r < nums.size() - 1 && nums[r + 1] == target)
                    ++r;
                res.push_back(l);
                res.push_back(r);
                break;
            }
            else if(nums[mid] < target)
                l = mid + 1;
            else
                r = mid - 1;
        }
        if(res.empty())
            return {-1,-1};
        return res;
    }
};