LeetCode笔记-----在排序数组中查找元素的第一个和最后一个位置

写在前边

本文为系列文章,记录自己在leetcode上刷题的一些笔记,欢迎大家一块儿讨论。

题目描述

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

解决思路

这道题目要求时间复杂度是O(logn)也就是在提示我们不能单纯的顺序遍历数组,应该减少循环的次数,那就不得不提二分法,恰好之前做过一个二分查找的题目,那我们就先看一下二分查找的例子:

int search(int* nums, int numsSize, int target) {
    int low = 0;
    int high = numsSize-1;
    int mid;
    while(low<=high){
        mid = (low+high)/2;
        if(nums[mid] == target)
            return mid;
        if(nums[mid]<target)
            low = low+1;
        else
            high = high-1;
    }
    return -1;
}

二分查找的思想不难理解,但跟我们这道题有什么关联呢?最初的想法就是使用二分查找找到目标值的其中一个下标,再根据该下标向左向右分别遍历得到初始下标和结束下标。具体可看一下代码

class Solution {
    public int[] searchRange(int[] nums, int target) {
         int[] result = new int[]{-1, -1};
        int left = 0;
        int right = nums.length-1;
        while(left<=right){
            int mid = (left + right)/2;
            if(nums[mid]==target){
                while(mid>=left && nums[mid]==target){
                    mid--;
                }
                result[0] = mid+1;
                mid = (left + right)/2;
                while(mid<=right && nums[mid]==target){
                    mid++;
                }
                result[1] = mid - 1;
                break;
            }else if (nums[mid] > target){
                right = mid-1;
            }else{
                left = mid+1;
            }
        }
        return result;

    }
}

当然上边只是我的解法,但是保证是AC的,还有更为高端的解法,大家可以参考这个,个人感觉写的挺详细的,大家可以一块儿探讨探讨


代码啥的已经更新在GitHub上,欢迎大家指点github

猜你喜欢

转载自blog.csdn.net/liu0415111/article/details/81388104