栈是一种先进后出的数据结构,工作中我们很少用到这种数据结构,我了解到的只有jvm底层调用方法栈的时候有用到,其实我们java递归方法就是栈的一种应用。
由于最近在学数据结构和算法,小老弟我就去力扣上面做了几道栈的算法题,没做几道就把小老弟我难住了,题目如下
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 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; }