程序员面试的那些小事架构算法设计模式和编程理论程序员

lintcode 寻找峰值

2017-01-11  本文已影响107人  yzawyx0220

你给出一个整数数组(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 所在位置.
题目链接:http://www.lintcode.com/zh-cn/problem/find-peak-element/

二分查找法,如果mid比mid+1大,舍弃左边,如果mid比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
        int i = 0,j = A.size() - 1;
        while (i < j) {
            int mid = (i + j) / 2;
            if (A[mid] > A[mid + 1]) j = mid;
            else i = mid + 1;
        }
        return j;
    }
};
上一篇 下一篇

猜你喜欢

热点阅读