首页 > 代码库 > 【LeetCode】Search in Rotated Sorted Array
【LeetCode】Search in Rotated Sorted Array
Suppose a sorted array is rotated at some pivot unknown to you beforehand.
(i.e., 0 1 2 4 5 6 7
might become 4 5 6 7 0 1 2
).
You are given a target value to search. If found in the array return its index, otherwise return -1.
You may assume no duplicate exists in the array.
package com.leetcode.SearchInRotatedSortedArray; /** * (1)如果target==A[m],那么m就是我们要的结果,直接返回; (2)如果A[m]<A[r],那么说明从m到r一定是有序的(没有受到rotate的影响),那么我们只需要判断target是不是在m到r之间,如果是则把左边缘移到m+1,否则就target在另一半,即把右边缘移到m-1。 (3)如果A[m]>=A[r],那么说明从l到m一定是有序的,同样只需要判断target是否在这个范围内,相应的移动边缘即可。 * @author huanli2 * */ public class NSolution { public int search(int[] A, int target) { int start = 0; int end = A.length-1; while(end>=start){ int mid = (end+start)/2; if(A[mid]==target) return mid; if(A[mid]<A[end]){ if(A[mid]<target&&A[end]>=target) start=mid+1; else end=mid-1; }else{ if(A[mid]>target&&A[start]<=target) end=mid-1; else start=mid+1; } } return -1; } }
public class Solution { public int search(int[] A, int target) { if(target==A[0]) return 0; if(target>A[0]){ int pre = A[0]; for(int i=1;i<A.length;i++){ if(A[i]<pre) return -1; if(A[i]==target) return i; pre=A[i]; } } if(target<A[0]){ if(target==A[A.length-1]) return A.length-1; int pre = A[A.length-1]; for(int i=A.length-2;i>=0;i--){ if(A[i]>pre) return -1; if(A[i]==target){ return i; } pre=A[i]; } } return -1; } }
这道题是二分查找Search Insert Position的变体,看似有点麻烦,其实理清一下还是比较简单的。因为rotate的缘故,当我们切取一半的时候可能会出现误区,所以我们要做进一步的判断。具体来说,假设数组是A,每次左边缘为l,右边缘为r,还有中间位置是m。在每次迭代中,分三种情况:
(1)如果target==A[m],那么m就是我们要的结果,直接返回;
(2)如果A[m]<A[r],那么说明从m到r一定是有序的(没有受到rotate的影响),那么我们只需要判断target是不是在m到r之间,如果是则把左边缘移到m+1,否则就target在另一半,即把右边缘移到m-1。
(3)如果A[m]>=A[r],那么说明从l到m一定是有序的,同样只需要判断target是否在这个范围内,相应的移动边缘即可。
根据以上方法,每次我们都可以切掉一半的数据,所以算法的时间复杂度是O(logn),空间复杂度是O(1)。
public boolean search(int[] A, int target) { if(A==null || A.length==0) return false; int l = 0; int r = A.length-1; while(l<=r) { int m = (l+r)/2; if(A[m]==target) return true; if(A[m]>A[l]) { if(A[m]>target && A[l]<=target) { r = m-1; } else { l = m+1; } } else if(A[m]<A[l]) { if(A[m]<target && A[r]>=target) { l = m+1; } else { r = m-1; } } else { l++; } } return false; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。