查找无续数组的峰值
题目说明
给出一个整数数组(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 所在位置.
思路
一开始拿到题目觉得很简单,一个循环就好,1分钟写完代码
class Solution { public: /* * @param A: An integers array. * @return: return any of peek positions. */ int findPeak(vector<int>& A) { // write your code here unsigned length = A.size()-1; for (unsigned i = 1; i < length; ++i) { if (A[i] > A[i-1] && A[i] > A[i+1]) { return i; } } return -1; } };
提交之后,返回了一个红色的Time Limit Exceeded - -
- 经过仔细思考,可以按照二分法来做,这样时间复杂度可以降为O(logN),具体思路,这里盗用下别人的图:
中间值mid只会存在图中的四种情况:
- 处于峰值,这时候就是我们要找的位置,直接返回就好
- 处于最低点,则根据A[1] > A[0]判断,mid前面必有峰值
- 处于下坡,则根据A[1] > A[0]判断,mid前面必有峰值
- 处于上坡,则根据A[n-2] > A[n-1]判断,mid后面必有峰值
思路理清楚之后代码就很好解决了
class Solution { public: /* * @param A: An integers array. * @return: return any of peek positions. */ int findPeak(vector<int>& A) { // write your code here unsigned left = 0, right = A.size() - 1; while (left <= right) { unsigned mid = (left + right) / 2; if (A[mid-1] < A[mid]) { if (A[mid] < A[mid+1]) { // 上坡; left = mid; } else { // 最高点; return mid; } } else { right = mid; } } return -1; } };
总结
需要学会把复杂的问题进行拆分成各个小问题,然后一个个去解决。