题目
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。
搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。
你可以假设数组中不存在重复的元素。
你的算法时间复杂度必须是 O(log n) 级别。
示例 1:
输入: nums = [4,5,6,7,0,1,2], target = 0
输出: 4
示例 2:
输入: nums = [4,5,6,7,0,1,2], target = 3
输出: -1
我的思路
1、首先找到转折点,并返回其索引。
2、使用二分查找算法,在转折点两边的数据中进行查找。
我的代码
//
// Created by HINTS on 2018/12/17.
//
#include <iostream>
#include <vector>
using namespace std;
//二分查找算法实现
int binarySearch(vector<int> &nums, int l, int r, int target){
if(l > r)
return -1;
int mid = l + (r - l)/2;
if(nums[mid] == target)
return mid;
else if(nums[mid] < target){
return binarySearch(nums,mid + 1, r, target);
}else {
return binarySearch(nums, l, mid -1, target);
}
}
int search(vector<int> &nums, int target){
int tmp = 0; //记录转折点的位置
int first = nums[0];//数组第一个数据,用来和target进行比较,以确定在转折点哪边进行二分查找。
int result = 0;//返回结果
bool flag = true; //标志位,用来确定是否存在转折点,若不存在,则对nums全部数据进行二分查找。
for (int i = 0; i < nums.size() - 1; ++i) {
if(nums[i] > nums[i+1]){
flag = false;
tmp = i;
break;
}
}
if(flag){
result = binarySearch(nums, 0, nums.size()-1, target);
}else{
if(target >= first)
result = binarySearch(nums, 0, tmp, target);
else
result = binarySearch(nums, tmp+1, nums.size()-1, target);
}
return result;
}
int main033(){
vector<int> nums = {1,3};
int out = search(nums, 3);
cout << out <<endl;
return 0;
}
感悟
方法不止这一个,再接再厉。