尋找峯值

你給出一個整數數組(size爲n),其具有以下特點:

    相鄰位置的數字是不同的
    A[0] < A[1] 並且 A[n - 2] > A[n - 1]

假定P是峯值的位置則滿足A[P] > A[P-1]且A[P] > A[P+1],返回數組中任意一個峯值的位置。
注意事項

數組可能包含多個峯值,只需找到其中的任何一個即可

樣例

給出數組[1, 2, 1, 3, 4, 5, 7, 6]返回1, 即數值 2 所在位置, 或者6, 即數值 7 所在位置.


採用二分查找,由於題目中有個條件A[0] < A[1] 並且 A[n - 2] > A[n - 1],即數組的起始爲低值,二分查找判斷時,若A[mid] >A[mid-1],說明往mid-1左走是上坡,因此在mid-1一定能找到峯值。


class Solution {
public:
    /**
     * @param A: An integers array.
     * @return: return any of peek positions.
     */
    int findPeak(vector<int> A) {
        // write your code here
        if(A.empty())
          return 0;
          
        int l = 0;
        int r = A.size()-1;
        while(l < r)
        {
            int mid = (l+r)/2;
            if(mid == 0)
             return 1;
            if(mid == A.size()-1)
             return mid-1;
            if(A[mid] > A[mid+1] && A[mid] > A[mid-1])
             return mid;
            if(A[mid] < A[mid+1])
             l = mid+1;
            else
             r = mid; // 不是減1
        }
        return 0;
    }
};


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章