题目描述
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
解题思路
单调栈
public static int[] nextGreaterElements(int[] nums) {
int length = nums.length;
int res[] = new int[length];
Arrays.fill(res,-1); //默认都为1
Stack<Integer> stack = new Stack<>();
//相当于把数组遍历两遍
for (int i = 0; i < length * 2; i++) {
//遍历数组的第index(index从0开始)个元素,因为数组会遍历
//两遍,会导致数组越界异常,所以这里要对数组长度进行求余
int index = i % length;
//单调栈,他存储的是元素的下标,不是元素具体值,从栈顶
//到栈底所对应的值是递增的(栈顶元素在数组中对应的值最小,
//栈底元素对应的值最大),如果栈顶元素对应的值比nums[index]小,
//说明栈顶元素对应的值遇到了右边第一个比他大的值,然后栈顶元素出栈,
//让他对应的位置变为nums[index],也就是他右边第一个比他大的值,
//然后继续判断……
while (!stack.isEmpty() && nums[stack.peek()] < nums[index]){
res[stack.pop()] = nums[index];
}
//当前元素的下标入栈
stack.push(index);
}
return res;
}