题目描述(中等难度)

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

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

先分享 leetcode 提供的两个解法。

解法一 线性扫描

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

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

  1. public int[] searchRange(int[] nums, int target) {
  2. int[] targetRange = {-1, -1};
  3. // 从左到右扫描
  4. for (int i = 0; i < nums.length; i++) {
  5. if (nums[i] == target) {
  6. targetRange[0] = i;
  7. break;
  8. }
  9. }
  10. // 如果之前没找到,直接返回 [ -1 , -1 ]
  11. if (targetRange[0] == -1) {
  12. return targetRange;
  13. }
  14. //从右到左扫描
  15. for (int j = nums.length-1; j >= 0; j--) {
  16. if (nums[j] == target) {
  17. targetRange[1] = j;
  18. break;
  19. }
  20. }
  21. return targetRange;
  22. }

时间复杂度:O(n)。

空间复杂度:O(1)。

解法二 二分查找

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

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

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

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

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

此时虽然 mid 指向的值等于 target 了,但是我们要找的其实还在左边,为了达到 log 的时间复杂度,我们依旧是丢弃一半,我们需要更新 end = mid - 1,图示如下。

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

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

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

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

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

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

所以最后的代码就出来了。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. } else if (target < nums[mid]) {
  32. end = mid - 1;
  33. } else {
  34. start = mid + 1;
  35. }
  36. }
  37. ans[1] = end;
  38. return ans;
  39. }

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

空间复杂度:O(1)。

解法三

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

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

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

考虑下边的一种情况,如果我们找最左边等于 target 的,此时 mid 的位置已经是我们要找的了,而解法二更新成了 end = mid - 1,然后继续循环了,而此时我们其实完全可以终止了。只需要判断 nums[ mid - 1] 是不是小于 nums [ mid ] ,如果小于就刚好是我们要找的了。

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

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

  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. end = mid - 1;
  40. } else {
  41. start = mid + 1;
  42. }
  43. }
  44. return ans;
  45. }

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

空间复杂度:O(1)。

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

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

  1. if (target > n || mid == 0) {
  2. if (target < n || mid == nums.length - 1) {
  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. }
  40. return ans;
  41. }

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

windliang wechat

添加好友一起进步~

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

如果想系统的学习数据结构和算法,强烈推荐一个我之前学过的课程,可以点击 这里 查看详情