给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1] 输出: [2,-1,2] 解释: 第一个 1 的下一个更大的数是 2; 数字 2 找不到下一个更大的数; 第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。
思路:单调栈。
涉及到比较前后大小的问题,考虑单调栈解决。
对于此问题,用一个栈记录元素下标:
1.若栈不为空且当前元素>栈顶(下标)对应的元素,将res对应值赋值为当前元素,弹出栈顶元素(已经找到右边第一个大元素),进行下一次循环比较
2.否则,当前元素入栈
class Solution {
public int[] nextGreaterElements(int[] nums) {
int len=nums.length;
int[] res=new int[len];
Stack<Integer> stack=new Stack<>();
int[] nums2=new int[len*2];
// 初始化res
for(int i=0;i<len;i++) res[i]=-1;
// 复制新数组
for(int i=0;i<2;i++){
for(int j=0;j<len;j++){
nums2[i*len+j]=nums[j];
}
}
for(int i=0;i<len*2;i++){
while(!stack.isEmpty() && nums2[i]>nums2[stack.peek()]){
// 查看堆顶元素
res[stack.peek()]=nums2[i];
stack.pop();
}
if(i<len) stack.push(i);
}
return res;
}
}