《LeetCode之每日一题》:90.在排序数组中查找元素的第一个和最后一个位置

题目链接： 在排序数组中查找元素的第一个和最后一个位置

有关题目

给定一个按照升序排列的整数数组 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 <= 10^5
-10^9 <= nums[i] <= 10^9
nums 是一个非递减数组
-10^9 <= target <= 10^9

题解

法一：双指针

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> a = {-1,-1};
        bool left_flag = false,right_flag = false;
        int left =0,right =size(nums)-1; 
        while(right>=left){
            if(left_flag && right_flag) break;
            if(target == nums[left] && !left_flag){
                left_flag=true;
                a[0]=left;
            }
            if(target == nums[right] && !right_flag){
                right_flag=true;
                a[1]=right;
            }
            if(!left_flag) left++;//右移
            if(!right_flag) right--;//左移
        }
        return a;
    }
};

时间复杂度：O(n)
空间复杂度：O(1),返回数组大小不计入额外空间开辟

法二：二分查找
代码一：
（不知道的为啥，会超时）

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* searchRange(int* nums, int numsSize, int target, int* returnSize){
    int cnt1 = -1,cnt2 = -1;
    int l = 0, r = numsSize - 1;
    int* ret = malloc(sizeof(int) * 2);
    *returnSize = 2;
    while(l <= r){
        int mid = l + ((r - l) >> 1);
        if (nums[mid] == target){
            cnt1 = cnt2 = mid;
            for (; cnt1 >= 0 && nums[cnt1] == target; ){
            cnt1--;
            }
            
            for (; cnt2 < numsSize && nums[cnt2] == target; ){
            ++cnt2;
            }
            
        } else if (nums[mid] > target){
            r = mid - 1;
        } else {
            l = mid + 1;
        }
    }
    ret[0] = cnt1, ret[1] = cnt2;
    return ret;
}

代码二：

class Solution {
public:
    int l_Boundary(vector<int>& nums, int target){
        int l = 0, r = nums.size();
        while (l < r){//[l, r)
            int mid = l + (r - l) / 2;
            if (nums[mid] >= target){
                r = mid;//右区间取不到,所以不写 r = mid - 1;
            }
            else {
                l = mid + 1;
            }
        }
        if (l >= nums.size() || nums[l] != target){
            return -1;
        }
        return l;
    }
    int r_Boundary(vector<int>& nums, int target){
        int l = 0, r = nums.size();
        while(l < r){
            int mid = l + (r - l) / 2;
            if (nums[mid] <= target){
                l = mid + 1;
            }
            else 
            {
                r = mid;
            }
        }
        if (r <= 0 || nums[l - 1] != target){
            return -1;
        }
        return l - 1;//由于l = mid + 1且在该条件下， mid向着nums[mid] > target方向移动
    }
    vector<int> searchRange(vector<int>& nums, int target) {
        if (nums.size() == 0) return {-1, -1};
        int l_b = l_Boundary(nums, target);
        int r_b = r_Boundary(nums, target);
        if (l_b == -1 || r_b == -1){
            return {-1, -1};
        }
        return {l_b, r_b};
    }
};

代码三：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int binarySearch(int* nums, int numsSize, int target, bool lower){
     int left = 0, right = numsSize - 1, ans = numsSize;
    while (left <= right) {
        int mid = (left + right) / 2;
        if (nums[mid] > target || (lower && nums[mid] >= target)) {//细节之处，成功的增强代码的可复用性
        //nums[mid] > target--》查找第一个大于target的数组索引
        //lower && nums[mid] >= target--》查找第一个大于等于target的数组索引
            right = mid - 1;
            ans = mid;
        } else {
            left = mid + 1;
        }
    }
    return ans;

}
int* searchRange(int* nums, int numsSize, int target, int* returnSize) {
    int leftIdx = binarySearch(nums, numsSize, target, true);
    int rightIdx = binarySearch(nums, numsSize, target, false) - 1;//我们这边找到的是，第一个大于target的数的下标减去1
    int* ret = malloc(sizeof(int) * 2);
    *returnSize = 2;
    if (leftIdx <= rightIdx && rightIdx < numsSize && nums[leftIdx] == target && nums[rightIdx] == target) {
        ret[0] = leftIdx, ret[1] = rightIdx;
        return ret;
    }
    ret[0] = -1, ret[1] = -1;
    return ret;
}

补充：

题目：排序数组中查找出等于target元素个数。

代码一：

class Solution {
public:
    int search(vector<int>& nums, int target) {
        return upper_bound(begin(nums), end(nums), target) - lower_bound(begin(nums), end(nums), target);
        //upper_bound返回第一个大于target的下标
        //lower_bound返回第一个大于等于target的下标
    }
};

代码二：

class Solution {
public:
    int search(vector<int> const& nums, int target) {
        const auto [l, r] = equal_range(nums.begin(), nums.end(), target);
        
        return distance(l, r);
    }
    //函数equal_range()返回first和last之间等于val的元素区间
    //equal_range返回值为std::pair对象
    //其first和second成员都成为迭代器
	//first 成员的值等同于 std::lower_bound 执行于同一输入序列后的返回。
	//second 成员的值等同于 std::upper_bound 执行于同一输入序列后的返回。
	//如果nums中没有target，则返回[-1,-1];
    
};