看到时间复杂度要求为对数立刻想到二分法。注意有一个边界条件,在LintCode中的寻找峰值也有一个类似条件,本质是一样的,即中间的数两边,增长的一端一定有峰值。
class Solution {
public:
int findPeakElement(vector<int>& nums) {
if (nums.size()==1) return 0;
int head=0;
int tail=nums.size()-1;
while (head<tail)
{
int mid=(head+tail)/2;
if (nums[mid+1]>nums[mid]) head=mid+1;
else if (nums[mid-1]>nums[mid]) tail=mid-1;
else return mid;
}
return head;
}
};