0503.下一个更大的元素II
描述
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
实例
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意
输入数组的长度不会超过 10000。
题解
stack
存储尚未找到下一个更大元素的下标- 第一个循环遍历
nums
,对于每一个被遍历的元素- 如果栈不空且大于栈顶元素,则栈顶元素的下一个更大元素即为当前元素
- 出栈,并继续检查栈顶元素与当前元素的大小关系
- 否则将当前元素的下标入栈
- 如果栈不空且大于栈顶元素,则栈顶元素的下一个更大元素即为当前元素
- 同时在循环中找到
nums
中的最大数字 - 第二个循环,设置变量
i
作为当前比较的nums
中的元素下标- 如果栈顶元素为最大值,则没有下一个更大元素,对应位置填上
-1
并出栈 - 如果
nums[i]
大于栈顶元素,则将栈顶元素出栈,且下一个更大元素为nums[i]
- 否则
i++
- 如果栈顶元素为最大值,则没有下一个更大元素,对应位置填上
public static int[] nextGreaterElements(int[] nums) {
int[] result = new int[nums.length];
Stack<Integer> stack = new Stack<>();
int max = Integer.MIN_VALUE;
int nowNum;
for (int i=0;i<nums.length;i++){
nowNum = nums[i];
max = max>nowNum ? max:nowNum;
while (!stack.empty()){
if (nowNum > nums[stack.peek()]){
result[stack.pop()] = nowNum;
} else
break;
}
stack.push(i);
}
int i=0;
while (!stack.empty()){
if (nums[stack.peek()] == max){
result[stack.pop()] = -1;
continue;
}
if (nums[i] > nums[stack.peek()]){
result[stack.pop()] = nums[i];
continue;
}
i++;
}
return result;
}