程序员人生 网站导航

LeetCode Search in Rotated Sorted Array

栏目:互联网时间:2014-12-16 08:11:56

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 ⑴.

You may assume no duplicate exists in the array.

思路分析:这题一样考察2分查找,但是是在rotation以后的数组中进行2分查找。这题的特点是,每次2分后,最少有1半是有序的,另外一半是无序的(遭到了rotation的影响)。所以要利用这1特性,通过比较target和有序的那半边数组的最小数和最大数,来决定l和m的更新规则。具体而言

1.如果target = A[m] 返回m

2 如果target 不等于A[m]

2.1 如果A[m] < A[r],说明A[m...r]有序,没有收到rotation影响,可以判断target是不是在A[m...r]范围内,如果是更新l=m+1;否则更新r=m⑴

2.2 如果A[m] >= A[r],说明A[l...m]有序,没有收到rotation影响,可以判断target是不是在A[l...m]范围内,如果是更新r=m⑴;否则更新l=m+1

所以利用每次依然有序的半边数组,我们依然可以进行2分查找,更新l和r,每次缩小1半的搜索范围,时间复杂度O(log(n)),空间复杂度O(1),只有常数级别的额外空间。注意判断target是不是在A[m...r]范围和target是不是在A[l...m]范围内的时候,要斟酌target == A[r]和target == A[l]的情况,2分查找的实现等号情况的斟酌要特别谨慎。

AC 时间:5min。

AC Code

public class Solution { public int search(int[] A, int target) { //02:46 if(A == null || A.length == 0) return ⑴; int l = 0; int r = A.length - 1; while(l <= r){ int m = (l + r) / 2; if(target == A[m]) return m; else{ if(A[m] < A[r]){ //m ... r is sorted if(target <= A[r] && target > A[m]) { l = m + 1; } else { r = m - 1; } } else { //l ... m is sorted if(target >= A[l] && target < A[m]){ r = m ⑴; } else { l = m + 1; } } } } return ⑴; } //02:51 }


------分隔线----------------------------
------分隔线----------------------------

最新技术推荐