LeetCode演算法題33:搜尋旋轉排序陣列解析
阿新 • • 發佈:2019-01-03
假設按照升序排序的陣列在預先未知的某個點上進行了旋轉。
( 例如,陣列 [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
這裡不太清楚這個旋轉得意思,但是基本上來說應該等同於迴圈右移,所以這樣的話不管怎麼移陣列中總是有兩個有序數列,題目要求複雜度是O(logn),所以用的應該是二分法。二分法取的中間值判斷是否為目標值之後更新左右端點需要加一步,因為只有有序數列才能使用二分法更新端點,所以要找到中間值左側是有序數列還是右側是有序數列,這就比較簡單了,有兩種方式,第一種:看中間值如果大於左端點值,那麼左邊應該是有序得,否則就是右邊有序。第二種:看中間值如果小於右端點值,那麼右邊應該是有序得,否則就是左邊有序。在有序的一側進行端點更新即可。
C++原始碼:(第一種)
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0, right = nums.size()-1;
while(left <= right)
{
int mid = left + (right-left) / 2;
if(nums[mid]==target) return mid;
else if(nums[mid] >= nums[ left])
{
if(nums[left] <= target && nums[mid] > target)
right = mid - 1;
else
left = mid + 1;
}
else
{
if(nums[mid] < target && nums[right] >= target)
left = mid + 1;
else
right = mid - 1;
}
}
return -1;
}
};
python3原始碼:
class Solution:
def search(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
left = 0
right = len(nums) - 1
while left <= right:
mid = left + (right-left) // 2
if nums[mid]==target:
return mid
elif nums[mid] <= nums[right]:
if nums[mid] < target and nums[right] >= target:
left = mid + 1
else:
right = mid - 1
else:
if nums[left] <= target and nums[mid] > target:
right = mid - 1
else:
left = mid + 1
return -1