找到一个数组中每一个元素第一个比它大的元素

一、题目描述

        给定义个数组v,对于数组中每一个元素，找到其后面元素中第一个比它大的元素，并且记录在数组v2中返回。例如给定数组v = {1,3,6,-1,2},返回v2={3,6,-1,2,-1}。（如果找不到，记为-1），并且要求时间复杂度应该为O(N)。

二、解题思路

        可能看到这道题，大家首先会想到对于数组中的元素一个一个去遍历，进而找到第一个比它大的元素，但是这种做法的时间复杂度应该为O（n**2）,因此不可采用此种方法。

        进一步去思考，发现可以利用堆栈去解决，如果对于当前元素v[i],发现v[i]+1 < v[i],此时不用再去比较v[i+2]与v[i]的关系，而是把i压入堆栈中，去比较v[i+1]与v[i+2]的关系，一旦发现满足后一个元素大于第一个元素的，便输出。

vector<int> FindMax(vector<int> &num)
{
  int len = num.size();
if(len == 0) return {};
vector<int> res(len,-1);
stack<int>  notFound;
int i = 0;
while(i<len)
{
if(notFound.empty() || num[notFound.top()] > num[i])
{
    notFound.push(i++);
}
else
{
  res[notFound.top()] = num[i];

notFound.pop();
}

}
return res;

}