栈－leetcode 496. Next Greater Element I

最新推荐文章于 2021-03-09 17:08:40 发布

Maxwell__726

最新推荐文章于 2021-03-09 17:08:40 发布

阅读量547

点赞数

分类专栏： leetcode 文章标签： leetcode java 栈 hashmap

本文链接：https://blog.youkuaiyun.com/Maxwell__726/article/details/75195882

版权

leetcode 专栏收录该内容

51 篇文章

订阅专栏

原题链接：Next Greater Element I

题解：

public class Solution {
    public int[] nextGreaterElement(int[] findNums, int[] nums) {
        /*
            Time Complexity:O(N)
            Space Complexity:O(N)
        */        
        if(nums==null || nums.length==0)return new int[]{};
        HashMap<Integer,Integer>map=new HashMap<>();
        ArrayDeque<Integer>sta=new ArrayDeque<>();
        for(int num:nums){
            while(!sta.isEmpty() && sta.peek()<num){
                map.put(sta.pop(),num);
            }
            sta.push(num);
        }
        int[] res=new int[findNums.length];
        for(int i=0;i<findNums.length;i++){
            res[i]=map.getOrDefault(findNums[i],-1);
        }
        return res;        
    }
}