C++刷笔试题（寻找最后的山峰）【美团点评】【2020】

糖逗

题目描述：

山峰元素是指其值大于或等于左右相邻值的元素。给定一个输入数组nums，任意两个相邻元素值不相等，数组可能包含多个山峰。找到索引最大的那个山峰元素并返回其索引。

假设 nums[-1] = nums[n] = -∞。


输入描述:
在命令行中输入一行数字，数字之间以空格分割，遇到换行符结束。输入的数字为整型，且总数量在10万以内。



输出描述:
输出索引最大的山峰的索引值（一个数字）

示例1
输入
2 4 1 2 7 8 4
输出
5
说明
索引最大的山峰的高度为8，其索引为5

题目来源：牛客网

#include <vector>
#include <iostream>
using namespace std;
 
int solution(vector<int>&input){
    //其实元素位置和最后的元素位置单独讨论
    int len = input.size();
    if(input[len-1] >= input[len-2]) return len-1;
    int res;
    bool flag = false;
    for(int i = 1; i < input.size()-1; i++){
        if(input[i] >= input[i-1] && input[i] >= input[i+1]){
            res = i;
            flag = true;
        }
    }
    if(flag == false) return 0;
    return res;
}
 
int main(void){
    int number;
    vector<int> input;
    while(cin >> number){
        input.push_back(number);
    }
    cout << solution(input) << endl;
    return 0;
}

糖逗