题目描述(中等难度)

找到目标值的第一次出现和最后一次出现的位置,同样要求 log ( n ) 下完成。

先分享 提供的两个解法。

解法一 线性扫描

从左向右遍历,一旦出现等于 target 的值就结束,保存当前下标。如果从左到右没有找到 target,那么就直接返回 [ -1 , -1 ] 就可以了,因为从左到右没找到,那么从右到左也一定不会找到的。如果找到了,然后再从右到左遍历,一旦出现等于 target 的值就结束,保存当前下标。

时间复杂度是 O(n)并不满足题意,但可以了解下这个思路,从左到右,从右到左之前也遇到过。

时间复杂度:O(n)。

空间复杂度:O(1)。

解法二 二分查找

让我们先看下正常的二分查找。

  1. int end = nums.length - 1;
  2. while (start <= end) {
  3. int mid = (start + end) / 2;
  4. if (target == nums[mid]) {
  5. return mid;
  6. } else if (target < nums[mid]) {
  7. end = mid - 1;
  8. } else {
  9. start = mid + 1;
  10. }
  11. }

二分查找中,我们找到 target 就结束了,这里我们需要修改下。

我们如果找最左边等于 target 的值,找到 target 时候并不代表我们找到了我们所需要的,例如下边的情况,

34. Find First and Last Position of Element in Sorted Array - 图1

此时 tartget > nums [ mid ] ,更新 start = mid + 1。

此时 target == nums [ mid ] ,但由于我们改成了 end = mid - 1,所以继续更新,end 就到了 mid 的左边,此时 start > end 了,就会走出 while 循环, 我们要找的值刚好就是 start 指向的了。那么我们修改的代码如下:

找右边的同样的分析思路,就是判断需要丢弃哪一边。

所以最后的代码就出来了。leetcode 中是把找左边和找右边的合并起来了,本质是一样的。

  1. public int[] searchRange(int[] nums, int target) {
  2. int start = 0;
  3. int end = nums.length - 1;
  4. int[] ans = { -1, -1 };
  5. if (nums.length == 0) {
  6. return ans;
  7. }
  8. while (start <= end) {
  9. int mid = (start + end) / 2;
  10. if (target == nums[mid]) {
  11. end = mid - 1;
  12. } else if (target < nums[mid]) {
  13. end = mid - 1;
  14. } else {
  15. start = mid + 1;
  16. }
  17. }
  18. //考虑 tartget 是否存在,判断我们要找的值是否等于 target 并且是否越界
  19. if (start == nums.length || nums[ start ] != target) {
  20. return ans;
  21. } else {
  22. ans[0] = start;
  23. }
  24. ans[0] = start;
  25. start = 0;
  26. end = nums.length - 1;
  27. while (start <= end) {
  28. int mid = (start + end) / 2;
  29. if (target == nums[mid]) {
  30. start = mid + 1;
  31. end = mid - 1;
  32. } else {
  33. }
  34. }
  35. ans[1] = end;
  36. return ans;
  37. }

时间复杂度:O(log(n))。

空间复杂度:O(1)。

解法三

以上是 leetcode 提供的思路,我觉得不是很好,因为它所有的情况都一定是循环 log(n)次,讲一下我最开始想到的。

相当于在解法二的基础上优化了一下,下边是解法二的代码。

34. Find First and Last Position of Element in Sorted Array - 图2

当然,找最右边也是同样的思路,看下代码吧。

  1. public int[] searchRange(int[] nums, int target) {
  2. int start = 0;
  3. int end = nums.length - 1;
  4. int[] ans = { -1, -1 };
  5. if (nums.length == 0) {
  6. return ans;
  7. }
  8. while (start <= end) {
  9. int mid = (start + end) / 2;
  10. if (target == nums[mid]) {
  11. //这里是为了处理 mid - 1 越界的问题,可以仔细想下。
  12. //如果 mid == 0,那么 mid 一定是我们要找的了,而此时 mid - 1 就会越界了,
  13. //为了使得下边的 target > n 一定成立,我们把 n 赋成最小值
  14. //如果 mid > 0,直接吧 nums[mid - 1] 赋给 n 就可以了。
  15. int n = mid > 0 ? nums[mid - 1] : Integer.MIN_VALUE;
  16. if (target > n) {
  17. ans[0] = mid;
  18. break;
  19. }
  20. end = mid - 1;
  21. } else if (target < nums[mid]) {
  22. end = mid - 1;
  23. } else {
  24. start = mid + 1;
  25. }
  26. }
  27. start = 0;
  28. end = nums.length - 1;
  29. while (start <= end) {
  30. int mid = (start + end) / 2;
  31. if (target == nums[mid]) {
  32. int n = mid < nums.length - 1 ? nums[mid + 1] : Integer.MAX_VALUE;
  33. if (target < n) {
  34. ans[1] = mid;
  35. break;
  36. }
  37. start = mid + 1;
  38. } else if (target < nums[mid]) {
  39. start = mid + 1;
  40. }
  41. }
  42. return ans;
  43. }

时间复杂度:O(log(n))。

空间复杂度:O(1)。

@JZW 的提醒下,上边的虽然能 AC,但是如果要找的数字刚好就是 Integer.MIN_VALUE 的话,就会出现错误。可以修改一下。

主要是这两句,除了小于 n,还判断了当前是不是在两端。

  1. public int[] searchRange(int[] nums, int target) {
  2. int start = 0;
  3. int end = nums.length - 1;
  4. int[] ans = { -1, -1 };
  5. if (nums.length == 0) {
  6. return ans;
  7. }
  8. while (start <= end) {
  9. int mid = (start + end) / 2;
  10. if (target == nums[mid]) {
  11. int n = mid > 0 ? nums[mid - 1] : Integer.MIN_VALUE;
  12. if (target > n || mid == 0) {
  13. ans[0] = mid;
  14. break;
  15. }
  16. end = mid - 1;
  17. } else if (target < nums[mid]) {
  18. end = mid - 1;
  19. } else {
  20. start = mid + 1;
  21. }
  22. }
  23. start = 0;
  24. end = nums.length - 1;
  25. while (start <= end) {
  26. int mid = (start + end) / 2;
  27. if (target == nums[mid]) {
  28. int n = mid < nums.length - 1 ? nums[mid + 1] : Integer.MAX_VALUE;
  29. if (target < n || mid == nums.length - 1) {
  30. ans[1] = mid;
  31. break;
  32. }
  33. start = mid + 1;
  34. } else if (target < nums[mid]) {
  35. end = mid - 1;
  36. } else {
  37. start = mid + 1;
  38. }
  39. }

总体来说,这道题并不难,本质就是对二分查找的修改,以便满足我们的需求。

添加好友一起进步~

如果觉得有帮助的话,可以点击 这里 给一个 star 哦 ^^