1. 程式人生 > 實用技巧 >leetcode33 搜尋旋轉排序陣列(Medium)

leetcode33 搜尋旋轉排序陣列(Medium)

題目來源:leetcode33 搜尋旋轉排序陣列

題目描述:

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

( 例如,陣列 [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

解題思路:

使用二分法。先要判斷mid在哪個有序的部分,在分別二分。

class Solution {
public:
    int search(vector<int>& nums, int target) {
        if(nums.size()<1) return -1;
        int l=0,r=nums.size()-1;
        if(r==0) return nums[0]==target?0:-1;//只包含一個數的情況
        int mid=l+(r-l)/2;
        while(l<=r){
            mid=l+(r-l)/2;
            if(target==nums[mid]) return mid;
            //start~mid是有序陣列
            if(nums[mid]>=nums[r]){
                if(target>=nums[l]&&target<nums[mid]){
                    r=mid-1;
                }
                else{
                    l=mid+1;
                }
            }
            //mid~end是有序陣列
            else{
                if(nums[mid]<target&&target<=nums[r]){
                    l=mid+1;
                }
                else{
                    r=mid-1;
                }
            }
        }
        return -1;
    }
};