題目來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array
假設按照升序排序的數組在預先未知的某個點上進行了旋轉。
( 例如,數組 [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
該題是二分法查找的變種,但是萬變不離其宗,只是在中間多加一些判斷條件
class Solution {
public:
int search(vector<int>& nums, int target) {
if (nums.size() == 0) return -1;
return midSearch(nums, 0, nums.size()-1, target);
}
int midSearch(vector<int>& nums, int start_index, int end_index, int target)
{
if (nums[start_index] == target) return start_index;
if (nums[end_index] == target) return end_index;
if (end_index-start_index<2)
{
return -1;
}
int mid = (start_index+end_index)/2;
if (nums[mid] == target) return mid;
//mid->end_index是連續排序的數組,反轉出現在start_index->mid之間
if (nums[mid] < nums[end_index])
{
if (target > nums[mid] && target < nums[end_index])
{
return midSearch(nums, mid, end_index, target);
}
else
{
return midSearch(nums, start_index, mid, target);
}
}
else //if (nums[start_index] < nums[mid]) //start_index->mid是連續排序的數組,反轉出現在 mid->end_index之間
{
if (target < nums[mid] && target > nums[start_index])
{
return midSearch(nums, start_index, mid, target);
}
else
{
return midSearch(nums, mid, end_index, target);
}
}
};
};
結果如下:(效果還有待改進,可是運行了官方的和一些精選的代碼,也是這個結果,不知道怎麼回事!!!)