思路
看到这道题的时候我不假思索地写出了这样的代码:
/** * 寻找最后的山峰 * @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; }