503. 下一个更大元素 II
题目
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
解题思路
和496. 下一个更大元素 I题目相似,就是变成了环而已,所以我们只需要把数组翻倍,尾部的元素就可以和自己左边的元素进行比较了。所以只需要
①翻倍数组
②用下一个更大元素的模板
代码
class Solution {
public int[] nextGreaterElements(int[] nums) {
int[] res=new int[nums.length];
Deque<Integer> stack = new LinkedList<>();
int n=nums.length;
//把数组翻倍
for(int i=2*n-1;i>=0;i--){//i=2*n-1:相当于逻辑上把数组扩大一倍,比如原来是{1,2,3}现在就是{1,2,3,1,2,3}
//原来下标是2,扩大一倍后,还想取到该元素就5%6=5,下标为5,值也是3.用%是为了防止索引越界
while(!stack.isEmpty()&&stack.peek()<=nums[i%n]){
stack.pop();//弹出比nums[i]小的元素
}
res[i%n]=stack.isEmpty()?-1:stack.peek();
stack.push(nums[i%n]);
}
return res;
}
}