题解 | #寻找峰值#

思路

看到这道题的时候我不假思索地写出了这样的代码：

/**
 * 寻找最后的山峰
 * @param a int整型一维数组 
 * @param aLen int a数组长度
 * @return int整型
 */
int solve(int* a, int aLen ) {
    // write code here
        int i;
    int index = 0;
    for (i=0; i<aLen-2; i++)
    {
        if (a[i+1] >= a[i] && a[i+1]>=a[i+2])
        {
            index = i+1;
        }
    }
    return index;
}

这段代码在特定的条件下是对的，比如样例中的数组：[2,4,1,2,7,8,4]。

但是这段代码没有考虑任何边界的条件，比如[2,9]，期望的值应该是1，即9这个元素的下标。

因此这段代码需要加上一些判断，绝对不能像我这样还贴心的跳过了最后俩元素。

下面的代码就是正解了，除去两个特殊情况的判断外，倒序遍历数组，只要遇到峰值就直接输出：

/**
 * 寻找最后的山峰
 * @param a int整型一维数组 
 * @param aLen int a数组长度
 * @return int整型
 */
int solve(int* a, int aLen ) {
    // write code here
    int i;
    int index = 0;
    for (i=aLen-1; i>=0; i--)
    {
        if (i == aLen - 1)
        {
            if (a[i]>=a[i-1])
            {
                return i;
            }
        }
        else if (i == 0)
        {
            if (a[i] >= a[i+1])
            {
                return i;
            }
        }
        else 
        {
            if (a[i] >= a[i-1] && a[i] >= a[i+1])
            {
                return i;
            }
        }
    }
    return -1;
}

这段代码改一改，就能完成寻找第一个峰值的逻辑：

/**
 * 寻找最后的山峰
 * @param a int整型一维数组 
 * @param aLen int a数组长度
 * @return int整型
 */
int solve(int* a, int aLen ) {
    // write code here
    int i;
    int index = 0;
    for (i = 0; i < aLen; i++)
    {
        if (i==0)
        {
            if (a[i]>=a[i+1])
            {
                return i;
            }
        }
        else if (i==aLen-1)
        {
            if (a[i]>=a[i-1])
            {
                return i;
            }
        }
        else
        {
            if (a[i]>=a[i-1] && a[i]>=a[i+1])
            {

                return i;
            }
        }
    }
    return -1;
}