[leetcode] 34. 在排序数组中查找元素的第一个和最后一个位置(Java)

Posted 2020-12-04 ACBingo

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

题目要求用O（logn），明显要用二分。

其实二分不难，难的是要处理好边界

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int i = 0, j = nums.length;
        int mid = (i + j) / 2;

        int p = -1;
        while (i < j) {
            if (nums[mid] == target) {
                p = mid;
                break;
            }
            if (nums[mid] > target) {
                if (j == mid) break;
                j = mid;
                mid = (i + j) / 2;
            } else {
                if (i == mid) break;
                i = mid;
                mid = (i + j) / 2;
            }
        }

        if (p == -1) {
            return new int[]{-1, -1};
        } else {
            int a = p, b = p;
            while (a > 0 && nums[a - 1] == target) a--;
            while (b < nums.length - 1 && nums[b + 1] == target) b++;
            return new int[]{a, b};
        }

    }
}