LeetCode 34 在排序数组中查找元素的第一个和最后一个位置(二分查找)

  • 1. 题目链接
https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/
  • 2. 题目描述

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

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

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

示例 1:

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

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
  • 3. 题目分析

       根据题目要求,算法时间复杂度必须是 O(log n) 级别,我们选择使用二分查找算法实现,使用[left, right]代表整个序列的下标区间。

        左下标元素值 nums[left] 和有下标元素值 nums[right] 同时等于target时,即满足条件;

        左下标元素值 nums[left] 小于target时,left加一;

        右下标元素值 nums[right] 大于target时,right减一。

  • 4. 代码实现
int* searchRange(int* nums, int numsSize, int target, int* returnSize){
    int left = 0, right = numsSize - 1;
    int *res = (int *)malloc(sizeof(int) * 2);
    *returnSize = 2;
    res[0] = -1;
    res[1] = -1;

    while (left <= right) {
        if (nums[left] == target && nums[right] == target) {
            res[0] = left;
            res[1] = right;
            return res;
        }
        if (nums[left] < target)
            left++;
        if (nums[right] > target)
            right--;
    }
    return res;
}
  • 5. 调试报错

第一次调试时,用while (left < right)语句,会有用例报错,如下图所示,因为不加等于时,会将一个元素的用例情况,判断为不满足条件,和题意不符。

LeetCode 34 在排序数组中查找元素的第一个和最后一个位置(二分查找)_第1张图片

  • 6. 提交记录

LeetCode 34 在排序数组中查找元素的第一个和最后一个位置(二分查找)_第2张图片

你可能感兴趣的:(leetcode)