
文 | 肴尧谣
来源:图解面试算法「ID: LeetCodeAnimation」

题目来源于 LeetCode 上第 34 号问题:find-first-and-last-position-of-element-in-sorted-array。题目难度为 中等。
题目描述
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。
示例:
输入: nums = [5,7,7,8,8,10], target = 8输出: [3,4]
输入: nums = [5,7,7,8,8,10], target = 6输出: [-1,-1]
题目解析
题目中要求了时间复杂度为O(log n),这就很清楚要使用二分查找法了。
首先定义两个指针变量,分别存储左右两个位置的索引。首先去找目标值的最左面的索引,通过循环为了防止元素丢失,每次保留最右面的元素,左侧的指针移动时+1。在循环结束的时候判断一下数组中是否包括目标值,不包括的话直接退出。右面的跟左侧相同,只不过正好相反。
动画描述
代码实现
// 34. 下一个排列// https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/// 时间复杂度:O(n)// 空间复杂度:O(1)class Solution {public int[] searchRange(int[] nums, int target) {int[] res = new int[] { -1, -1 };int left = 0;int right = nums.length - 1;int l = left;int r = right;while (left < right) {int mid = (left + right) / 2;if (nums[mid] < target) {left = mid + 1;} else {right = mid;}}if (left>right||nums[left]!=target) {return new int[]{-1,-1};}while (l < r) {int mid = (l + r) / 2 + 1;if (nums[mid] > target) {r = mid - 1;} else {l = mid;}}if (left > right || left > r) {return new int[] { -1, -1 };} else {return new int[] { left, r };}}}
老规矩,兄弟们还记得么,右下角的 “在看” 点一下,如果感觉文章内容不错的话,记得分享朋友圈让更多的人知道!
每晚 8 点,不见不散,一起刷 LeetCode !



原文始发于微信公众号(五分钟学算法):LeetCode 图解 | 34.在排序数组中查找元素的第一个和最后一个位置




