码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode:Find Peak Element

时间:2014-12-24 13:32:13      阅读:119      评论:0      收藏:0      [点我收藏+]

标签:algorithm   leetcode   

题目描述:

A peak element is an element that is greater than its neighbors.

Given an input array where num[i] ≠ num[i+1], find a peak element and return its index.

The array may contain multiple peaks, in that case return the index to any one of the peaks is fine.

You may imagine that num[-1] = num[n] = -∞.

For example, in array [1, 2, 3, 1], 3 is a peak element and your function should return the index number 2.

思路:采用二分的思想。如果num[mid]>num[mid+1]并且num[mid]>num[mid-1],则直接返回mid;如果num[mid]>num[mid+1]并且num[mid]<num[mid-1],则说明peak元素在数组的左半部分;如果num[mid]<num[mid+1]并且num[mid]>num[mid-1],则说明peak元素在数组的右半部分。


代码:

int findPeak(const vector<int> &num,int start,int end)
{
    if(start == end)
        return start;
    else if(end - start == 1)
    {
        if(num[end] > num[start])
            return end;
        else
            return start;
    }
    else
    {
        int mid = (start+end)/2;
        if(num[mid] > num[mid+1] && num[mid] > num[mid-1])
            return mid;
        else if(num[mid] > num[mid+1] && num[mid-1] > num[mid])
            return findPeak(num,mid+1,end);
        else
            return findPeak(num,start,mid-1);
    }
}

int Solution::findPeakElement(const vector<int> &num)
{
        return findPeak(num,0,num.size()-1);
}




LeetCode:Find Peak Element

标签:algorithm   leetcode   

原文地址:http://blog.csdn.net/yao_wust/article/details/42122119

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!