【LeetCode】 34. Find First and Last Position of Element in Sorted Array （Medium）（JAVA）

【LeetCode】 34. Find First and Last Position of Element in Sorted Array 在排序数组中查找元素的第一个和最后一个位置（Medium）（JAVA）

题目地址: https://leetcode.com/problems/find-first-and-last-position-of-element-in-sorted-array/

题目描述：

Given an array of integers nums sorted in ascending order, find the starting and ending position of a given target value.

Your algorithm’s runtime complexity must be in the order of O(log n).

If the target is not found in the array, return [-1, -1].

Example 1:

Input: nums = [5,7,7,8,8,10], target = 8
Output: [3,4]

Example 2:

Input: nums = [5,7,7,8,8,10], target = 6
Output: [-1,-1]

题目大意

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

你的算法时间复杂度必须是 O(log n) 级别。

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

解题方法

1、看到你的算法时间复杂度必须是 O(log n) 级别，而且是有序数组，就知道要用二分查找
2、找到第一个位置
3、找到最后一个位置

note: 区分第一还是最后一个，就是通过 nums[mid] == target，放在 start 还是放在 end

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int start = 0;
        int end = nums.length - 1;
        while (start <= end) {
            int mid = start + (end - start) / 2;
            if (nums[mid] >= target) {
                end = mid - 1;
            } else {
                start = mid + 1;
            }
        }
        int[] res = new int[]{-1, -1};
        if (start >= nums.length || nums[start] != target) return res;
        res[0] = start;
        end = nums.length - 1;
        while (start <= end) {
            int mid = start + (end - start) / 2;
            if (nums[mid] > target) {
                end = mid - 1;
            } else {
                start = mid + 1;
            }
        }
        res[1] = end;
        return res;
    }
}

执行用时 : 0 ms, 在所有 Java 提交中击败了 100.00% 的用户
内存消耗 : 42.8 MB, 在所有 Java 提交中击败了 40.86% 的用户