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

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

已解答
中等
相关标签
相关企业
给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

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

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

示例 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：

输入：nums = [], target = 0
输出：[-1,-1]

提示：

0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109

https://leetcode.cn/problems/find-first-and-last-position-of-element-in-sorted-array/?envType=problem-list-v2&envId=9Rkf2tiI

思路：
1.如果找到的数小于目标值，它左边的范围就可以排除。
2.如果找到的数大于目标值，它右边的范围就可以排除。
3.在中间数和中间数之前只可能存在目标值的开始位置，在它之后只可能存在它的结束位置。
4.所以要分别找开始位置和结束位置。

/**
* Note: The returned array must be malloced, assume caller calls free().
*/

// 当找到的中间元素严格小于或大于目标值，它左边或右边的范围就可以排除
/* 在mid和mid之前一定只存在目标元素的开始位置 \
     * 找给定元素的开始位置 */
int findMidFirst(int* nums, int numsSize, int target) {
    int left = 0, right = numsSize - 1;
    while (left < right)
    {
        int mid = left + (right - left) / 2;
        if (nums[mid] < target)
            // 下一轮查找区间：[mid+1,right]
            left = mid + 1;
        else
            // 下一轮查找区间：[left,mid]
            right = mid;
    }
    if (nums[left] == target)
        return left;
    return -1;
}

/* 在mid和mid之后一定只存在目标元素的结束位置 \
     * 找给定元素的结束位置 */
int findMidLast(int* nums, int numsSize, int target) {
    int left = 0, right = numsSize - 1;
    while (left < right)
    {
        int mid = left + (right - left + 1) / 2;
        if (nums[mid] > target)
            // 下一轮：[left,mid-1]
            right = mid - 1;
        else
            // 下一轮：[mid,right]
            left = mid;
    }
    if (nums[left] == target)
        return left;
    return -1;
}

int* searchRange(int* nums, int numsSize, int target, int* returnSize) {
    if (numsSize == 0) {
        int* result = (int *)malloc(2 * sizeof(int));
        result[0] = -1;
        result[1] = -1;
        *returnSize = 2;         // 将要返回的空间设置成2个int大小
        return result; 
        }
    int* result = (int *)malloc(2 * sizeof(int));
    result[0] = findMidFirst(nums, numsSize, target);
    result[1] = findMidLast(nums, numsSize, target);
    *returnSize = 2;
    return result; 
}