题目描述:
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
输入输出样例:
示例 1:
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。
思路:
题目很简单,就是将要判断的的值放在一个容器里面,然后遍历数组对容器中的值全部进行判断,达成条件后将判断值从容器取出,判断完后将当前指向的数组元素放入容器,因容器中大一点的值遇到的第一个更大的值不一定是小一点的值遇到的第一个更大值,故采用单调栈来求解。
以下为代码:
class Solution
{
public:
vector<int> nextGreaterElements(vector<int>& nums)
{
stack<int> s;
int n=nums.size();
vector<int> ans(n,-1);
for(int i=0;i<n*2;i++)
{
while((!s.empty())&&(nums[s.top()]<nums[i%n]))
{
ans[s.top()]=nums[i%n];
s.pop();
}
if(i<n) s.push(i);
}
return ans;
}
};
总结:题目没太多好说的,但是代码太久没写了很多细节都忘了,这里稍微记录以下。
vector常用用法:头文件#include<vector>
新建时初始化:
vector<int> ivec(10,-1);//10和int元素,每个都初始化为-1
vector<int> ivec1(10);//10个int元素,每个初始化为0
常用操作:
v.empty();//检查是否为空
v.capacity();//返回当前vector中最大可以存储数据的容量
v.size();//返回v中元素的个数
v[n];//获取v中第n个元素
v.push_back(x);//添加元素x
stack栈的常用用法:头文件#include<stack>
常用操作:
s.push();//向栈内压入一个成员;
s.pop();//从栈顶弹出一个成员(与jave不一样,没有返回值);
s.empty();//如果栈为空返回true,否则返回false;
s.top();//返回栈顶,但不删除成员;
s.size();//返回栈内元素的大小;