Code & Fun

打卡,第16天

失眠的感觉真难受。。。一天都想睡觉,但是却睡不着,sad。

Suppose an array sorted in ascending order 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.

在有序的数组中进行查找,显然第一个想到的就是二分查找啦,不过题目给的数组是多了一个限定条件,就是这个数组被rotated了,所以,显然直接用二分查找是不行的。

观察4 5 6 7 0 1 2,我们可以发现如果我们可以找到7这个位置,我们就可以得到两个有序数组,可以进行二分查找,所以一个简单直观的想法就是:

1
2
3
4
5
6
7
if (nums[first] > nums[last]) {
while(last >= 0 && nums[first] > nums[last] && nums[last] < target)
last--;
while(first <= last && nums[first] > nums[last] && nums[first] > target)
first++;
}
//binarySearch

但是这样的时间复杂度就是O(n)了,显然不是我们想要的结果,我们可以对这个转折点进行一次二分查找:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
if(nums[first] > nums[last]) {
int f= first,l = last;
//找转折点
while(f <= l) {
mid = (f + l)/2;
if (nums[mid] > nums[mid + 1]) break;
else if (nums[mid] > nums[first]) f = mid + 1;
else if (nums[mid] < nums[last]) l = mid;
}
if (target > nums[last]) last = mid;
else if (target < nums[first]) first = mid + 1;
else if (target == nums[first]) return first;
else return last;
}
//binary Serarch

这样的时间复杂度就是O(2*logn)了。

完整代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
int search(vector<int>& nums, int target) {
int first = 0,last = nums.size() - 1;
int mid;

if (last < 0) return -1;

if(nums[first] > nums[last]) {
int f= first,l = last;
//找转折点
while(f <= l) {
mid = (f + l)/2;
if (nums[mid] > nums[mid + 1]) break;
else if (nums[mid] > nums[first]) f = mid + 1;
else if (nums[mid] < nums[last]) l = mid;
}
if (target > nums[last]) last = mid;
else if (target < nums[first]) first = mid + 1;
else if (target == nums[first]) return first;
else return last;
}
cout << first << last;
//binary search
while(first <= last) {
int mid = (first + last)/2;
if (nums[mid] == target) return mid;
else if (nums[mid] < target) first = mid + 1;
else last = mid - 1;
}
return -1;
}

不过显然这不是最优的方法啦,毕竟要O(2*logn),事实上只需要对二分查找进行修改,就可以直接运用了,恩,这是在dicuss中看到的方法:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
public int search(int[] A, int target) {
int lo = 0;
int hi = A.length - 1;
while (lo < hi) {
int mid = (lo + hi) / 2;
if (A[mid] == target) return mid;

if (A[lo] <= A[mid]) {
if (target >= A[lo] && target < A[mid]) {
hi = mid - 1;
} else {
lo = mid + 1;
}
} else {
if (target > A[mid] && target <= A[hi]) {
lo = mid + 1;
} else {
hi = mid - 1;
}
}
}
return A[lo] == target ? lo : -1;
}

本文首发于Code & Fun