利用单调栈解决 leetcode 496. 下一个更大元素 I

单调栈，顾名思义，指的是在栈中数字进行单调排列。
在这道题使用单调栈，数字进行递减排列。

这道题在leetcode上的链接
https://leetcode-cn.com/problems/next-greater-element-i/

下面贴一个单调栈的模板

vector<int> nextGreaterElement(vector<int>& nums) {
    
    
    vector<int> ans(nums.size()); // 存放答案的数组
    stack<int> s;
    for (int i = nums.size() - 1; i >= 0; i--) {
    
     // 倒着往栈里放
        while (!s.empty() && s.top() <= nums[i]) {
    
     // 判定个子高矮
            s.pop(); // 矮个起开，反正也被挡着了。。。
        }
        ans[i] = s.empty() ? -1 : s.top(); // 这个元素身后的第一个高个
        s.push(nums[i]); // 进队，接受之后的身高判定吧！
    }
    return ans;
}

作者：labuladong
链接：https://leetcode-cn.com/problems/next-greater-element-i/solution/dan-diao-zhan-jie-jue-next-greater-number-yi-lei-w/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

然后是我基于上面的模板的解决方案

class Solution {
    
    
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
    
      
        int[] result = new int[nums1.length];
        Map<Integer, Integer> map = nextGreaterHelper(nums2);      
        for (int i = 0; i < nums1.length; i++) {
    
    
            result[i] = map.get(nums1[i]);
        }
        return result;
    }

    private Map<Integer, Integer> nextGreaterHelper(int[] nums) {
    
    
        Map<Integer, Integer> map = new HashMap<>();
        Stack<Integer> stack = new Stack<>();  // 存放高个元素的栈
        for (int i = nums.length - 1; i >= 0; i--) {
    
      // 倒着往栈里放
            while (!stack.isEmpty() && stack.peek() <= nums[i]) {
    
    
                stack.pop();  // 矮个起开，反正也被挡着了
            }

            map.put(nums[i], stack.isEmpty() ? -1 : stack.peek());  // 当前元素身后的第一个高个
            stack.push(nums[i]);  // 进队，接受之后的身高判定
        }

        return map;
    }
}