单调栈,力扣题解-」更大元素 II

    技术2022-07-10  125

    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; }

     

    Processed: 0.013, SQL: 9