Google/LintCode：M-寻找峰值

题目

题目来源：Link

你给出一个整数数组(size为n)，其具有以下特点：

• 相邻位置的数字是不同的
• A[0] < A[1] 并且 A[n - 2] > A[n - 1]

假定P是峰值的位置则满足A[P] > A[P-1]且A[P] > A[P+1]，返回数组中任意一个峰值的位置。

 注意事项

数组可能包含多个峰值，只需找到其中的任何一个即可

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出数组[1, 2, 1, 3, 4, 5, 7, 6]返回1, 即数值 2 所在位置, 或者6, 即数值 7 所在位置.

分析

（1）方法一：直接遍历，TC=O(n)

（2）方法二：二分查找，TC=O(lgn)

代码

class Solution {
    /**
     * @param A: An integers array.
     * @return: return any of peek positions.
     */
     //遍历查找
     //TC=O(n)
    public int findPeak_n(int[] A) {
        // write your code here
        int n = A.length;
        
        for(int i=1; i<=n-2; i++){
            if(A[i-1]<A[i] && A[i]>A[i+1])
                return i;
        }
        
        return -1;
    }
    
    //二分查找
    //TC=(lgn)
    public int findPeak(int[] A) {
        // write your code here
        int n = A.length;
        
        find(A,1,n-2);
        
        return res;
    }
    int res=-1;
    void find(int[] A, int s, int e){
        if(res>0)
            return;
        
        if(s>e)
            return;
            
        int m = (s+e)/2;
        if(A[m-1]<A[m] && A[m]>A[m+1]){
            res=m;
            return;
        }
        
        find(A, s, m-1);
        if(res>0) return;
        find(A, m+1, e);
    }
}

改进的二分查找：

利用的是题目条件中的特性，首尾满足峰值条件，所以可以按照递增方向进行二分

class Solution {
    public int findPeak(int[] nums){
    	int s=1;
    	int e=nums.length-2;
    	while(s<e){
    		int m = (s+e)/2;
    		
    		if(nums[m]>nums[m-1] && nums[m]>nums[m+1])
    			return m;
    		else if(nums[m-1]>nums[m])
    			e = m-1;
    		else
    			s = m+1;
    	}
    	
    	return s;//s=e
    }
}