zoukankan      html  css  js  c++  java
  • Leetcode 503.下一个更大元素

    下一个更大元素

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

    示例 1:

    输入: [1,2,1]

    输出: [2,-1,2]

    解释: 第一个 1 的下一个更大的数是 2;

    数字 2 找不到下一个更大的数;

    第二个 1 的下一个最大的数需要循环搜索,结果也是 2。

    注意: 输入数组的长度不会超过 10000。

    思路

    使用栈来进行优化上面的算法,我们遍历两倍的数组,然后还是坐标i对n取余,取出数字,如果此时栈不为空,且栈顶元素小于当前数字,说明当前数字就是栈顶元素的右边第一个较大数,那么建立二者的映射,并且去除当前栈顶元素,最后如果i小于n,则把i压入栈。因为res的长度必须是n,超过n的部分我们只是为了给之前栈中的数字找较大值,所以不能压入栈,参见代码如下:

     1 import java.util.Arrays;
     2 import java.util.Stack;
     3 
     4 class Solution {
     5     public int[] nextGreaterElements(int[] nums) {
     6         int n=nums.length;
     7         int[] res=new int[nums.length];
     8         Arrays.fill(res,-1);
     9         Stack<Integer> stack=new Stack<Integer>();
    10         for(int i=0;i<2*n;i++){
    11             int num=nums[i%n];
    12             while(!stack.isEmpty()&&nums[stack.peek()]<num){
    13                 res[stack.peek()]=num;
    14                 stack.pop();
    15             }
    16             if(i<n) stack.push(i);
    17         }
    18         return res;
    19     }
    20 }
  • 相关阅读:
    Java Web系统经常使用的第三方接口
    Direct UI
    Python 分析Twitter用户喜爱的推文
    数据挖掘十大经典算法(9) 朴素贝叶斯分类器 Naive Bayes
    利用Excel批量高速发送电子邮件
    普林斯顿大学数学系的崛起
    Node.js学习
    映射 SQL 和 Java 类型
    Nutch配置
    OGNL
  • 原文地址:https://www.cnblogs.com/kexinxin/p/10372517.html
Copyright © 2011-2022 走看看