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

亡梦爱人 提交于 2019-12-27 14:11:16

给定一个按照升序排列的整数数组 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]

解析:二分法的应用,注意边界的判断

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size();
        int mid;
        vector<int> res(2,-1);
        //寻找左边界
        while(left<right){    
            mid=left+(right-left)/2;
            if(nums[mid]>=target) right=mid;
            else left = mid+1;
        }
        if(left==nums.size()||nums[left]!=target) return {-1,-1};
        res[0]=left;
        left=0;
        right=nums.size();
        //寻找右边界
        while(left<right){    
            mid=left+(right-left)/2;
            if(nums[mid]<=target) left=mid+1;
            else right=mid;
        }
        if(nums[right-1]==target) res[1]=right-1;
        return res;
    }
};
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!