1.栈的简介
栈是一种先进后出的数据结构,工作中我们很少用到这种数据结构,我了解到的只有jvm底层调用方法栈的时候有用到,其实我们java递归方法就是栈的一种应用。
由于最近在学数据结构和算法,小老弟我就去力扣上面做了几道栈的算法题,没做几道就把小老弟我难住了,题目如下
2.栈的应用
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例如下
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
题目链接如下
https://leetcode-cn.com/problems/next-greater-element-ii
当时我想的一种思路是将这个数组遍历两次,根据当前下标的元素时候比当前下标之后的元素的值小,如果有的话就是当前下标下一个更大的值,思路如下图
但是怎么想也跟栈挂不起关系,百度了一下有种栈叫单调栈,就是站内的元素要么单调递增,要么就是单调递减;
原来栈还能这么用
思路大致如下
1.我们将数组从后往前遍历
2.如果栈顶有值,我们比较当前元素是否大于栈顶的值,
如果大于或等于,则将栈顶的元素取出,再进行步骤2操作
如果小于,则进行下一步操作
3.如果栈内为空,我们则给-1,如果不为空,将栈顶的元素记录下来,为当前元素的下一个更大的数
4.将该元素放入栈中
实现代码如下
public static int[] nextGreaterElements(int[] nums) {
Stack<Integer> stack = new Stack<>();
int[] fianlNums = new int[nums.length];
//1.我们将数组从后往前遍历
for (int i = 2*nums.length-1;i >= 0;i--){
//取模是为了获取当前元素的下标
//2.如果栈顶有值,我们比较当前元素是否大于栈顶的值
// 如果大于或等于,则将栈顶的元素取出,再进行步骤2操作
// 如果小于,则进行下一步操作
while (!stack.isEmpty()&&nums[i%nums.length]>=stack.peek())
stack.pop();
//3.如果栈内为空,我们则给-1,如果不为空,将栈顶的元素记录下来,为当前元素的下一个更大的数
if(stack.isEmpty()){
fianlNums[i % nums.length] = -1;
}else{
fianlNums[i % nums.length] = stack.peek();
}
//4.将该元素放入栈中
stack.push(nums[i%nums.length]);
}
return fianlNums;
}