java数据结构与算法

直接在java中一直使用原生的Stack作为栈,看到大佬们都改用Deque了,因此使用Deque作为栈做一个练习。

这是leetCode一个题目,我直接截个图吧

 

 

本题目可以分解为两个问题:
1.如何寻找某个元素右侧比他更大的第一个元素,答案是使用单调栈
2.如何存储第一个问题的结果,答案是哈希表
废话少说,上代码
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Map<Integer, Integer> map = new HashMap<>();
        Deque<Integer> stack = new ArrayDeque<>();
        int n = nums1.length;
        int m = nums2.length;
        int[] ans = new int[n];
        //初始化单调栈,栈中是比nums2[i]大的元素的按升序保存
        for (int i = m - 1; i >= 0; i--) {
            //栈中有小于当前元素的元素,全部弹出
            int num = nums2[i];
            while (!stack.isEmpty() && stack.peek() <= num) {
                stack.pop();
            }
            map.put(num, stack.isEmpty() ? -1 : stack.peek());
            stack.push(num);
        }

        for (int i = 0; i < n; i++) {
            ans[i] = map.get(nums1[i]);
        }
        return ans;
    }

 

posted @ 2022-10-18 15:20  Mars.wang  阅读(35)  评论(0编辑  收藏  举报