题目描述(中等难度)

33. Search in Rotated Sorted Array - 图1

开始的时候想复杂了,其实就是一个排序好的数组,把前边的若干的个数,一起移动到末尾就行了。然后在 log (n) 下找到给定数字的下标。

总的来说,log(n),我们肯定得用二分的方法了。

解法一

参考这里-Binary-search-solution)首先我们想一下变化前,正常的升序。我们怎么找给定的数字。

33. Search in Rotated Sorted Array - 图2

我们每次只关心中间位置的值(这一点很重要),也就是上图 3 位置的数值,如果 target 小于 3 位置的值,我们就把 3 4 5 6 抛弃。然后看新的中间的位置,也就是 1 位置的数值。 3 位置, 1 位置的值是多少呢?我们有一个数组。

33. Search in Rotated Sorted Array - 图3

3 位置的值,刚好就是数组下标为 3 的值,1 位置的值刚好就是下标为 1 的值。

那么如果,按题目要求的,变化后,3 位置 和 1 位置的值怎么求呢? 此时我们的数组变成下边这样,我们依旧把值从小到大排列。

33. Search in Rotated Sorted Array - 图4

此时 3 位置的数值对应为数组下标是 0 的值,1 位置的值对应数组下标是 5 的值。任意位置的对应规则是什么呢?0 -> 4, 1 - > 5,4 ->1,就是就是 (位置 + 偏移 )% 数组的长度。这里就是加上 4 模 7。

问题转换为怎么去求出这个偏移。

我们只要知道任意一个位置对应的数组下标就可以了,为了方便我们可以求位置为 0 的值对应的下标(数组中最小的数对应的下标),0 位置对应的下标就是我们要求的偏移了(0 + 偏移 = 数组下标)。这里 nums = [ 4, 5, 6, 7, 0, 1, 2] ,我们就需要去求数值 0 的下标。

求最小值的下标,因为题目要求时间复杂度是 O(log ( n )),所以我们必须采取二分的方法去找,二分的方法就要保证每次比较后,去掉一半的元素。这里我们去比较中点和端点值的情况,那么是根据中点和起点比较,还是中点和终点比较呢?我们来分析下。

  • mid 和 start 比较

    mid > start : 最小值在左半部分。

    33. Search in Rotated Sorted Array - 图5

    mid < start: 最小值在左半部分。

    33. Search in Rotated Sorted Array - 图6

    无论大于小于,最小值都在左半部分,所以 mid 和 start 比较是不可取的。

  • mid 和 end 比较

    mid < end:最小值在左半部分。

    33. Search in Rotated Sorted Array - 图7

    33. Search in Rotated Sorted Array - 图8

    mid > end:最小值在右半部分。

    33. Search in Rotated Sorted Array - 图9

    所以我们只需要把 mid 和 end 比较,mid < end 丢弃右半部分(更新 end = mid),mid > end 丢弃左半部分(更新 start = mid)。直到 end 等于 start 时候结束就可以了。

但这样会有一个问题的,对于下边的例子,就会遇到死循环了。

33. Search in Rotated Sorted Array - 图10

问题出在,当数组剩偶数长度的时候,mid = (start + end)/ 2,mid 取的是左端点。上图的例子, mid > end, 更新 start = mid,start 位置并不会变化。那么下一次 mid 的值也不会变,就死循环了。所以,我们要更新 start = mid + 1。

综上,找最小值的下标的代码就出来了,同时,由于我们找的是位置 0 对应的下标,所以偏移就是最小值的下标。

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

当然,我们是找最小值对应的下标,然后求出了偏移。我们也可以找最大值的对应的下标,分析思路和之前是一样的,主要还是要注意一下边界的情况,然后就可以求出偏移。

  1. while (start < end) {
  2. int mid = Math.round(((float)start + end) / 2);
  3. if (nums[mid] < nums[start]) {
  4. end = mid - 1;
  5. } else {
  6. start = mid;
  7. }
  8. }
  9. int n = nums.length;
  10. bias = (start + n) - (n - 1); //此时 start 是最大值的数组下标,加上模长 n,减去最大值的位置 n - 1 ,就得到了偏移。因为 (位置 + 偏移)% n = 数组下标,即 (n - 1 + 偏移)% n = start,n - 1 加偏移超过了 n,所以取模理解成减 n 。

有了偏移,我们就可以愉快的找目标值的数组下标了。

  1. public int search (int[] nums, int target) {
  2. int start = 0;
  3. int end = nums.length - 1;
  4. //找出最小值的数组下标
  5. /* while (start < end) {
  6. int mid = (start + end) / 2;
  7. if (nums[mid] > nums[end]) {
  8. start = mid + 1 ;
  9. } else {
  10. end = mid;
  11. }
  12. }
  13. int bias = start;*/
  14. //找出最大值的数组下标
  15. while (start < end) {
  16. int mid = Math.round(((float)start + end) / 2);
  17. if (nums[mid] < nums[start]) {
  18. end = mid - 1;
  19. } else {
  20. start = mid;
  21. }
  22. }
  23. int n = nums.length;
  24. int bias = (start + n) - (n - 1); //得到偏移
  25. start = 0;
  26. end = nums.length - 1;
  27. while (start <= end) {
  28. int mid = (start + end) / 2;//中间的位置
  29. int mid_change = (mid + bias) % nums.length;//中间的位置对应的数组下标
  30. int value = nums[mid_change];//中间位置的值
  31. if (target == value) {
  32. return mid_change;
  33. }
  34. if (target < value) {
  35. end = mid - 1;
  36. } else {
  37. start = mid + 1;
  38. }
  39. }
  40. return -1;
  41. }

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

空间复杂度:O(1)。

解法二

参考这里,题目中的数组,其实是两段有序的数组。例如

[ 4 5 6 7 1 2 3 ] ,[ 4 5 6 7 ] 和 [ 1 2 3 ] 两段有序。

而对于 [ 1 2 3 4] 这种,可以看做 [ 1 2 3 4 ] 和 [ ] 特殊的两段有序。

而对于我们要找的 target , target 不在的那一段,所有数字可以看做无穷大,这样整个数组就可以看做有序的了,可以用正常的二分法去找 target 了,例如

[ 4 5 6 7 1 2 3] ,如果 target = 5,那么数组可以看做 [ 4 5 6 7 inf inf inf ]。

[ 4 5 6 7 1 2 3] ,如果 target = 2,那么数组可以看做 [ -inf -inf - inf -inf 1 2 3]。

和解法一一样,我们每次只关心 mid 的值,所以 mid 要么就是 nums [ mid ],要么就是 inf 或者 -inf。

什么时候是 nums [ mid ] 呢?

当 nums [ mid ] 和 target 在同一段里边。

  • 怎么判断 nums [ mid ] 和 target 在同一段?

    把 nums [ mid ] 和 target 同时与 nums [ 0 ] 比较,如果它俩都大于 nums [ 0 ] 或者都小于 nums [ 0 ],那么就代表它俩在同一段。例如

    [ 4 5 6 7 1 2 3],如果 target = 5,此时数组看做 [ 4 5 6 7 inf inf inf ]。nums [ mid ] = 7,target > nums [ 0 ],nums [ mid ] > nums [ 0 ],所以它们在同一段 nums [ mid ] = 7,不用变化。

  • 怎么判断 nums [ mid ] 和 target 不在同一段?

    把 nums [ mid ] 和 target 同时与 nums [ 0 ] 比较,如果它俩一个大于 nums [ 0 ] 一个小于 nums [ 0 ],那么就代表它俩不在同一段。例如

    [ 4 5 6 7 1 2 3],如果 target = 2,此时数组看做 [ - inf - inf - inf - inf 1 2 3]。nums [ mid ] = 7,target < nums [ 0 ],nums [ mid ] > nums [ 0 ],一个大于,一个小于,所以它们不在同一段 nums [ mid ] = - inf,变成了负无穷大。

看下代码吧

  1. public int search(int[] nums, int target) {
  2. int lo = 0, hi = nums.length - 1;
  3. while (lo <= hi) {
  4. int mid = lo + (hi - lo) / 2;
  5. int num = nums[mid];
  6. //nums [ mid ] 和 target 在同一段
  7. if ((nums[mid] < nums[0]) == (target < nums[0])) {
  8. num = nums[mid];
  9. //nums [ mid ] 和 target 不在同一段,同时还要考虑下变成 -inf 还是 inf。
  10. } else {
  11. num = target < nums[0] ? Integer.MIN_VALUE : Integer.MAX_VALUE;
  12. }
  13. if (num < target)
  14. lo = mid + 1;
  15. else if (num > target)
  16. hi = mid - 1;
  17. else
  18. return mid;
  19. }
  20. return -1;
  21. }

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

空间复杂度:O(1)。

解法三

参考这里,算法基于一个事实,数组从任意位置劈开后,至少有一半是有序的,什么意思呢?

比如 [ 4 5 6 7 1 2 3] ,从 7 劈开,左边是 [ 4 5 6 7] 右边是 [ 7 1 2 3],左边是有序的。

基于这个事实。

我们可以先找到哪一段是有序的 (只要判断端点即可),然后看 target 在不在这一段里,如果在,那么就把另一半丢弃。如果不在,那么就把这一段丢弃。

  1. public int search(int[] nums, int target) {
  2. int start = 0;
  3. int end = nums.length - 1;
  4. while (start <= end) {
  5. int mid = (start + end) / 2;
  6. if (target == nums[mid]) {
  7. return mid;
  8. }
  9. //左半段是有序的
  10. if (nums[start] <= nums[mid]) {
  11. //target 在这段里
  12. if (target >= nums[start] && target < nums[mid]) {
  13. end = mid - 1;
  14. } else {
  15. start = mid + 1;
  16. }
  17. //右半段是有序的
  18. } else {
  19. if (target > nums[mid] && target <= nums[end]) {
  20. start = mid + 1;
  21. } else {
  22. end = mid - 1;
  23. }
  24. }
  25. }
  26. return -1;
  27. }

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

空间复杂度:O(1)。

三种解法是从不同的思路去理解题意,但本质上都是找到丢弃一半的规则,从而达到 log (n) 的时间复杂度,对二分查找的本质的理解更加深刻了。

windliang wechat

添加好友一起进步~

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

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