力扣 503. 下一个更大元素 II

题目说明

给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。

  • 输入数组的长度不会超过 10000。

示例

例1

1
2
3
4
5
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。

例2

1
2
输入: [5,4,3,2,1]
输出: [-1,5,5,5,5]

笔者理解

此题是一道数组算法问题,在力扣题库中被定义为中等题。

解法

当笔者阅读完此题后,发现此题运用栈比较合适,当我们遍历到一个数时,就将其插入栈中,如果当前元素比栈首元素要大时,我们将其视为一个组合,然后将栈首元素弹出。让我们来看看具体如何实现的吧。

实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
public int[] nextGreaterElements(int[] nums) {
int n = nums.length;
int[] result = new int[n];
Arrays.fill(result, -1);
//都预设其没有后继更大数

Deque<Integer> stack = new LinkedList<Integer>();
//用LinkedList来实现Deque双端队列

for (int i = 0; i < n * 2 - 1; i++) {
while (!stack.isEmpty() && nums[stack.peek()] < nums[i % n]) {
//队列不为空qie队列首元素小于当前遍历元素

result[stack.pop()] = nums[i % n];
//将首元素对应的下标元素的值置为当前发现的首个大于其元素的值
//并从队列中删除
}
stack.push(i % n);
//放置下标
}
return result;
}

时间和空间效率都还行,可见此解法还比较适合此题;

image.png

总结

本题是今天的每日一题,难度是为中等,感兴趣的朋友都可以去尝试一下,此题还有其他更多的解法,朋友们可以自己逐一尝试。