1. 程式人生 > >Leetcode33.Search in Rotated Sorted Array搜尋旋轉排序陣列

Leetcode33.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)
    {
        int len = nums.size();
        int low = 0;
        int high = len - 1;
        //找到正常排序的起點,即原陣列的第一個元素
        while(low < high)
        {
            int mid = (low + high) / 2;
            //如果用<來判斷,那麼對應下面會改為high = mid - 1,在正常的排序下high最後會<0
            if(nums[mid] > nums[high])
            {
                low = mid + 1;
            }
            else
            {
                high--;
            }
        }
        int rotate = low;

        low = 0;
        high = len - 1;
        while(low <= high)
        {
            int mid = (low + high) / 2;
            //將mid值進行旋轉
            int realMid = (mid + rotate) % len;
            if(nums[realMid] == target)
                return realMid;
            else if(nums[realMid] > target)
            {
                //必須在原mid值上進行操作
                high = mid -1;
            }
            else
            {
                low = mid + 1;
            }
        }
        return -1;
    }
};