力扣496.下一个更大的元素I

力扣

这题目我也是用了两种解法来解题。

        第一种解法：因为该题其实可以看做是从后往前去找比目标元素更大的数组，而Stack后入先出的特性就可以很方便的处理这个问题。

        具体是用两个Stack，先用一个Stack1存储nums2数组中的元素，之后拿num1数组中的元素来nums2中查找结果，再定义一个对象来记录每次找到比num1中指定元素更大的元素。

        但是每次都应该从nums2从头往后找，所以创建了另一个Stack2来装从Stack1pop出来的元素，每次查找完后，再将Stacke2中的元素，重新放回Stack1中。代码：

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Stack<Integer> stack = new Stack<>();
        int[] ans = new int[nums1.length];
        for (Integer i : nums2) {
            stack.push(i);
        }
        for (int i = 0; i < nums1.length; i++) {
            boolean flag = false;
            Stack<Integer> temp = new Stack<>();
            int max = -1;
            while (!stack.isEmpty() && flag == false){
                int p = stack.pop();
                if(p > nums1[i]){
                    max = p;
                }
                else if(p == nums1[i]){
                    flag = true;
                }
                temp.push(p);
            }
            ans[i] = max;
            while (!temp.empty()){
                stack.push(temp.pop());
            }
        }
        return ans;
    }
}

        第二种解法，用Stack和HashMap来解决该题，用Stack来装nums2中的元素，HashMap的key表示nums2中的元素，而value则表示离key最近并且比key更大的元素。

        首先遍历数组，将nums2中的元素存储于Stack，之后进行第二次循环的时候，如果nums2的元素比Stack中最顶端的元素更大，那么就记录于HashMap中，直到Stack为空或者Stack中的值比数组元素大为止，这样就能够将nums2中的每个元素以及比他更大元素(若没有则为-1)存储于HashMap中。

        之后再用num1中的元素再HashMap中进行查找，就能得到结构。代码：

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
         int[] res = new int[nums1.length];
        Stack<Integer> stack = new Stack<>();
        HashMap<Integer, Integer> map = new HashMap<>();
        for (Integer i : nums2) {
            while (!stack.isEmpty() && i > stack.peek()) {
                int temp = stack.pop();
                map.put(temp, i);
            }
            stack.push(i);
        }
        while (!stack.isEmpty()) {
            int temp = stack.pop();
            map.put(temp, -1);
        }
        for (int i = 0; i < nums1.length; i++) {
            res[i] = map.get(nums1[i]);
        }
        return res;
    }
}