【LeetCode-中等】34. 在排序数组中查找元素的第一个和最后一个位置 - 数组双指针

34. 在排序数组中查找元素的第一个和最后一个位置

解法一:遍历

我的解法是先从左开始遍历数组,找到 nums 数组中第一个大于等于 target 的值,low为其下标。考虑到存在 nums = [2, 2],target = 3 这样的情况,判断条件还需要加上 low < high,防止数组越界。

找到第一个大于等于 target 的值后,判断该值是否等于 target ,如不等于,则表示该数组中不存在 target 值,返回 [-1, -1]. 若存在则用同样的思路从数组的最右侧逐步向左寻找第一个 target 值,用 high 表示其下标,最终结果即为[low, high].

class Solution {
    
    
public:
    vector<int> searchRange(vector<int>& nums, int target) {
    
    
        vector<int> result = {
    
    -1,-1};
        // nums为空
        if(nums.empty()) return result;
        int low = 0;
        int high = nums.size() - 1;
        // low < high : nums = [2, 2] target = 3
        while(nums[low] < target && low < high){
    
    
            low++;
        } 
        if(nums[low] != target) return result;
        while(nums[high] > target){
    
    
            high--;
        }
        result[0] = low;
        result[1] = high;
        return result;
    }
};

解法二:二分查找

这是力扣提供的题解,该题解充分利用了 nums 数组非递减排序的特点,采用二分查找的方法解决问题。

因为整个数组是非递减排序,可以使用二分法寻找第一个等于(或大于) target 的位置 leftIdx 和 第一个大于 target 的位置减一 rightIdx

由于 nums 数组中可能不存在 target ,因此得到的两个下标 leftIdx 和 rightIdx 进行校验,看是否符合条件,如果符合条件就返回 [leftIdx, rightIdx],否则返回 [-1,-1]。

class Solution {
    
     
public:
	// 二分查找,若 lower 为 true,查找第一个大于等于 target 的下标,为 false 则查找第一个大于 target 的下标
    int binarySearch(vector<int>& nums, int target, bool lower) {
    
    
        int left = 0, right = (int)nums.size() - 1, ans = (int)nums.size();
        while (left <= right) {
    
    
            int mid = (left + right) / 2;
            if (nums[mid] > target || (lower && nums[mid] >= target)) {
    
    
                right = mid - 1;
                ans = mid;
            } else {
    
    
                left = mid + 1;
            }
        }
        return ans;
    }
	
    vector<int> searchRange(vector<int>& nums, int target) {
    
    
        int leftIdx = binarySearch(nums, target, true);
        int rightIdx = binarySearch(nums, target, false) - 1;
        if (leftIdx <= rightIdx && rightIdx < nums.size() && nums[leftIdx] == target && nums[rightIdx] == target) {
    
    
            return vector<int>{
    
    leftIdx, rightIdx};
        } 
        return vector<int>{
    
    -1, -1};
    }
};

猜你喜欢

转载自blog.csdn.net/qq_43619058/article/details/125874552