在做题中学习(49):排序数组中查找元素的第一个和最后一个位置

34. 在排序数组中查找元素的第一个和最后一个位置 - 力扣(LeetCode)

题目要求:时间复杂度为:O(logN),而且是有序数组,那一定就是二分法解决了

解法:二分查找

1.左端点

要找一个元素的第一次出现的位置的情况,我们可以把数组分为两个区间,如下:

因为如果mid 落在 <3的区间,可以直接让left = mid + 1,因为答案绝对不可能在 <3 的区间里。

1. mid < target left = mid + 1;

因为如果mid 落在 >=3的区间,就不可以让 right = mid - 1,因为此区间是 >= 3的,所以有可能mid就在此区间第一个 == 3的位置,如果更新right 为 mid - 1,那么就会跑到<3的区间,永远都找不到值。

2.mid >= target right = mid;

3.循环结束条件 while(left < right) 因为上面的逻辑 最后会走到 left == right 才会有最终答案

2.右端点

要找一个元素的最后一次出现的位置的情况,我们可以把数组分为两个区间,如下:

因为如果mid 落在 >3的区间,可以直接让right = mid - 1,因为答案绝对不可能在 >3 的区间里。

1. mid > target right = mid - 1;

因为如果mid 落在 <=3的区间,就不可以让 left = mid + 1,因为此区间是 <= 3的,所以有可能mid就在此区间最后一个 == 3的位置,如果更新left 为 mid + 1,那么就会跑到>3的区间,永远都找不到值。

2.mid <= target left = mid;

3.循环结束条件 while(left < right) 因为上面的逻辑 最后会走到 left == right 才会有最终答案

细节问题

1.数组中没有target的情况

直接先在开始判断 :如果 数组元素为0,返回 {-1,-1}

判断左端点的时候 最后left == right ,只需循环结束后,判断一下 != target 返回{-1,-1}即可.

2.不动问题

求左端点, 如果用右边的 +1 公式求mid,那么最后剩两个元素的时候,如果命中 mid >= target的条件上,那么right不会动,程序陷入死循环, 所以要用不+1的公式。

求右端点, 如果用左边 不+1 公式求mid,那么最后剩两个元素的时候,如果命中 mid <= target的条件上,那么left不会动,程序陷入死循环, 所以要用+1的公式。

cpp 复制代码
class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) 
    {
        if(nums.size() == 0)
            return {-1,-1};
        vector<int> ret;
        int mid = 0;
        //1.寻找区间左端点
        int left = 0,right = nums.size()-1;
        while(left<right)
        {
            mid = left + (right - left) /2;
            if(nums[mid] < target) 
                left = mid + 1;
            else if(nums[mid] >= target)
                right = mid;
        }
        //走到这,left = right,找到左端点,但还需判断是不是 == target
        if(nums[left] != target)
            return {-1,-1};
        else
            ret.push_back(left);
        //2.寻找区间右端点
        left = 0,right = nums.size()-1;
        while(left<right)
        {
            mid = left + (right - left + 1) /2;
            if(nums[mid] <= target)
                left = mid;
            else if(nums[mid] > target)
                right = mid - 1;
        }
        //走到这,left = right,找到右端点
        ret.push_back(left);
        return ret;
    }
};
相关推荐
Elias不吃糖3 分钟前
LeetCode每日一练(3)
c++·算法·leetcode
小龙报20 分钟前
《算法通关指南数据结构和算法篇(2)--- 链表专题》
c语言·数据结构·c++·算法·链表·学习方法·visual studio
艾莉丝努力练剑40 分钟前
【优选算法必刷100题】第031~32题(前缀和算法):连续数组、矩阵区域和
大数据·人工智能·线性代数·算法·矩阵·二维前缀和
醉颜凉42 分钟前
环形房屋如何 “安全劫舍”?动态规划解题逻辑与技巧
c语言·算法·动态规划
mjhcsp43 分钟前
C++ 动态规划(Dynamic Programming)详解:从理论到实战
c++·动态规划·1024程序员节
大雨淅淅1 小时前
一文搞懂动态规划:从入门到精通
算法·动态规划
不去幼儿园1 小时前
【启发式算法】灰狼优化算法(Grey Wolf Optimizer, GWO)详细介绍(Python)
人工智能·python·算法·机器学习·启发式算法
随意起个昵称1 小时前
【二分】洛谷P2920,P2985做题小记
c++·算法
没书读了1 小时前
计算机组成原理-考前记忆清单
线性代数·算法
Hcoco_me1 小时前
大模型面试题5:矩阵(M*M)特征值分解的步骤
算法·机器学习·矩阵