题目:
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。
链接:https://leetcode-cn.com/problems/next-greater-element-ii
解题:
class Solution {
/*
解题思路:
计算机的内存都是线性的,没有真正意义上的环形数组,但是我们可以模拟出环形数组的效果,一般是通过 \%% 运算符求模(余数),获得环形特效:
int[] arr = {1,2,3,4,5};
int n = arr.length, index = 0;
while (true) {
print(arr[index % n]);
index++;
}
解题思路:利用一个辅助栈,遍历两次数组元素(实现循环),
第一次遍历时,将元素压入栈中,在每次遍历时,首先判断栈中是否有元素,当前元素是否大于栈顶元素,
若均符合则该元素即为栈顶元素的下一个更大的数。
循环数组:可以利用去余的方式循环访问数组即为循环数组结构了;
遍历次数为两遍数组即可,如果两边都没找到比该元素大的元素,那该元素没有比它大的即为-1;
*/
public static int[] nextGreaterElements(int[] nums) {
if (null == nums || nums.length == 0) {
return nums;
}
int length = nums.length;
int[] result = new int[length];
Arrays.fill(result, -1);
Stack<Integer> stack = new Stack<>();
// 遍历两遍循环数组
for (int i = 0; i < 2 * length - 1; i ++) {
int tem = i % length;
while (!stack.empty() && nums[tem] > nums[stack.peek()]) {
result[stack.pop()] = nums[tem];
}
// 元素放入
if (i < length) {
stack.push(i);
}
}
return result;
}
public static void main(String[] args) {
int[] arr = new int[]{1,2,1};
int[] result = nextGreaterElements(arr);
System.out.println(Arrays.toString(result));//[2, -1, 2]
}
}