1. 程式人生 > >旋轉數組找一個數

旋轉數組找一個數

-c ner pro ear 復雜 二分 detail content info

題目描述

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

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

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

你可以假設數組中不存在重復的元素。

你的算法時間復雜度必須是 O(log n) 級別。

思路:二分,二分之後的某一段必然是有序的。另一段必然是是部分有序的。如果落在有序的一段,直接二分找出該值。

如果是落在另一段部分有序的,繼續二分,該部分是遞歸的,有一段是有序的,另一段是部分有序的。

技術分享圖片

/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////l

from StefanPochmann https://leetcode.com/problems/search-in-rotated-sorted-array/discuss/14419/Pretty-short-C%2B%2BJavaRubyPython

用異或來判斷三段的升降序。

int search(vector<int>& nums, int target) {
    int lo = 0, hi = int(nums.size()) - 1;
    while (lo < hi) {
        int mid = (lo + hi) / 2;
        if ((nums[0] > target) ^ (nums[0] > nums[mid]) ^ (target > nums[mid]))
            lo = mid + 1;
        else
            hi = mid;
    }
    return lo == hi && nums[lo] == target ? lo : -1;
}

旋轉數組找一個數