java_leetcode刷题—单调栈

下一个更大元素I

（对应leetcode496）

由于刚开始学习数据结构以及java语言，所以参考其他人的答案，自己复述一遍再加上自己的理解写在这里加深印象。

方法：单调栈

题目给出两个数组，数组1是数组2的子集，且没有重复元素。希望在数组2中找到比数组1中的每一个元素要大的第一个元素。
可以利用单调栈，先单独看数组2，找到数组2中每个元素对应的下一个比他大的元素，两个元素保存在散列表中。如果没有对应的元素，按题目要求则对应着-1。由于数组1是数组2的子集，所以只需要在散列表中遍历数组1的元素即可。

单调栈

顾名思义单调栈中存储的数据是单调的，要么单调递增要么单调递减。如果希望栈中的数据是单调递减的，那么首先nums[0]压入栈中，nums[1]与nums[0]进行比较，如果nums[1]<nums[0],则将nums[1]压入栈中，否则将nums[1]弹出，继续比较nums[3],以此类推。

针对这道题，我们是要找到下一个比元素大的元素。首先将nums[0]入栈，如果nums[1]>nums[0],则找到了比nums[0]大的下一个元素，因此将这两个元素放入散列表中。将nums[1]压入栈中。继续看nums[2],如果nums[2]<nums[1]，将nums[2]压入栈中，继续看nums[3]，如果nums[3]>nums[1]，那么说明第四个元素比第二、三个元素都要大，那么就找到了第二、三个元素对应的下一个比他大的元素。接着将第四个元素nums[3]压入栈中以此类推。

public class Solution {
    public int[] nextGreaterElement(int[] findNums, int[] nums) {
        Stack < Integer > stack = new Stack < > ();//创建空栈
        HashMap < Integer, Integer > map = new HashMap < > ();//创建哈希表
        int[] res = new int[findNums.length];//创建最后返回的空列表
        for (int i = 0; i < nums.length; i++) {
            while (!stack.empty() && nums[i] > stack.peek())
                map.put(stack.pop(), nums[i]);
            stack.push(nums[i]);
        }//遍历数组2，将符合条件的元素放入哈希表
        while (!stack.empty())
            map.put(stack.pop(), -1);//找不到指定元素的元素，将他对应着-1放入哈希表
        for (int i = 0; i < findNums.length; i++) {
            res[i] = map.get(findNums[i]);
        }
        return res;
    }
}