1. 程式人生 > >LeetCode演算法題33:搜尋旋轉排序陣列解析

LeetCode演算法題33:搜尋旋轉排序陣列解析

假設按照升序排序的陣列在預先未知的某個點上進行了旋轉。
( 例如,陣列 [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