目录
思路
本题的解题思路主要利用栈的后进先出(LIFO)特性来找到数组中每个元素的下一个更大元素。
解题方法
1.创建一个与输入数组nums相同长度的结果数组result,并初始化为-1,表示如果没有找到下一个更大元素,则结果为-1。
2.创建一个栈stack,用于保存当前还未找到下一个更大元素的数组元素的索引。
3.遍历数组nums两次(或者将数组视为两倍长度的循环数组)。在第一次遍历中,我们只会将索引压入栈中,而在第二次遍历中(通过取模实现),我们会比较当前元素和栈顶元素所指的元素大小,并更新result数组。
4.在遍历过程中,如果当前元素大于栈顶元素所指的元素,那么栈顶元素所指的元素就找到了它的下一个更大元素,我们将这个元素从栈中弹出,并更新result数组。
5.遍历完成后,栈中剩余的元素在数组中都没有下一个更大元素,所以它们的结果保持为-1。
复杂度
时间复杂度:O(2n)=O(n)
空间复杂度:𝑂(𝑛)O(n)
Code
import java.util.ArrayDeque;
import java.util.Deque;
class Solution{
public int[] nextGreaterElements(int[] nums) {
int n = nums.length;
int[] result = new int[n];
Arrays.fill(result, -1); // 初始化结果数组为-1
Deque<Integer> stack = new ArrayDeque<>();
// 将nums数组两倍长度连接起来,以便循环搜索
for (int i = 0; i < 2 * n; i++) {
int num = nums[i % n]; // 通过取模实现循环
// 如果栈不为空且当前元素大于栈顶元素,更新栈顶元素的下一个更大元素
while (!stack.isEmpty() && num > nums[stack.peek()]) {
result[stack.pop()] = num;
}
// 将当前索引压入栈中
if (i < n) { // 只对前n个元素压栈,因为后n个元素只是为了循环搜索
stack.push(i);
}
}
return result;
}
}
路漫漫其修远兮,吾将上下而求索。——(屈原)