leetcode 34. 在排序數組中查找元素的第一個和最後一個位置 JAVA

題目:

給定一個按照升序排列的整數數組 nums,和一個目標值 target。找出給定目標值在數組中的開始位置和結束位置。算法

你的算法時間複雜度必須是 O(log n) 級別。數組

若是數組中不存在目標值,返回 [-1, -1]spa

示例 1:code

輸入: nums = [, target = 8
輸出: [3,4]5,7,7,8,8,10]

示例 2:blog

輸入: nums = [, target = 6
輸出: [-1,-1]5,7,7,8,8,10]

解題思路:

由於時間複雜度爲O(log n) 級別,因此必定使用到二分法。首先使用二分法找到target的位置,而後在target的位置上向前向後查找 get

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) {  //找到target的位置
            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 {  //在target所在位置向前向後查找
            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}; } }
相關文章
相關標籤/搜索