1. 程式人生 > 實用技巧 >LeetCode33:搜尋旋轉排序陣列

LeetCode33:搜尋旋轉排序陣列

假設按照升序排序的陣列在預先未知的某個點上進行了旋轉。

( 例如,陣列[0,1,2,4,5,6,7]可能變為[4,5,6,7,0,1,2])。

搜尋一個給定的目標值,如果陣列中存在這個目標值,則返回它的索引,否則返回-1。

你可以假設陣列中不存在重複的元素。

你的演算法時間複雜度必須是O(logn) 級別。

示例 1:

輸入: nums = [4,5,6,7,0,1,2], target = 0
輸出: 4
示例2:

輸入: nums = [4,5,6,7,0,1,2], target = 3
輸出: -1

對於一般的陣列,可以直接採用二分查詢;本題同樣也可以二分查詢,只是需要進行判斷:先找到有序的一側,如果target在這一側範圍內,就在這一側查詢,否則在另一側查詢。

 1 class Solution {
 2 public:
 3     int search(vector<int>& nums, int target) {
 4         if(nums.size()==0) return -1;
 5         int l=0,r=nums.size()-1,m=(l+r)/2;
 6         while(l<r-1){
 7             m=(l+r)/2;
 8             if(nums[l]<nums[m]){
 9                 if(nums[l]<=target && target<=nums[m]){
10 r=m; 11 } 12 else{ 13 l=m+1; 14 } 15 } 16 else if(nums[m]<nums[r]){ 17 if(nums[m]<=target && target<=nums[r]){ 18 l=m; 19 }
20 else{ 21 r=m-1; 22 } 23 } 24 } 25 if(target==nums[l]) return l; 26 if(target==nums[r]) return r; 27 return -1; 28 } 29 };