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

题目描述：

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

思路：

不难想到，第一次出现target的判定条件应该是
nums[index] == target && index == 0 || nums[index - 1] < target，
而最后一次出现的判定条件应该是
nums[index] == target && index == nums.length - 1 || nums[index + 1] > target.

所以这里可以分两次去找第一次出现和最后一次出现的位置

代码：

 class Solution {
    public int[] searchRange(int[] nums, int target) {
        int[] result = new int[2];
        result[0] = findTarget(nums, target, false);
        result[1] = findTarget(nums, target, true);
        return result;
    }
    
    // tag: false--第1次出现target  true--最后一次出现target
    public int findTarget(int[] nums, int target, boolean tag) {
        if (nums.length == 0 || nums[0] > target || nums[nums.length - 1] < target) {
            return -1;
        }
        int res = -1;
        int left = 0, right = nums.length - 1;
        int mid;
        while (left <= right) {
            mid = left + (right - left) / 2;
            if (nums[mid] == target) {
                if (!tag) {
                    right = mid - 1;
                    // 第1次出现target
                    if (mid == 0 || nums[mid - 1] < target) {
                        return mid;
                    }
                } else {
                    left = mid + 1;
                    // 最后一次出现target
                    if (mid == nums.length - 1 || nums[mid + 1] > target) {
                        return mid;
                    }
                }
            } else if (nums[mid] > target) {
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }
        return res;
    }
}

执行结果：通过
显示详情
执行用时 :1 ms, 在所有 Java 提交中击败了98.66%的用户
内存消耗 :38.9 MB, 在所有 Java 提交中击败了97.44%的用户

其实二分算法思想是很简单的，就是写的时候哟很多细节的地方需要注意，这个只能靠平常多联系多总结了。