代码随想录算法训练营 day58 | 739. 每日温度、496.下一个更大元素 I

代码随想录

739. 每日温度

思路

思路：从栈头到栈底（每次从栈头压入）升序的单调栈，一旦遇到更大的就先弹出再压入，否则直接压入。也就是每次必然会压入的。

代码

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int answer[] = new int[temperatures.length];

        Stack<Integer> s = new Stack<Integer>();
        s.push(0);
        for(int i = 1;i<temperatures.length;i++){
             while(!s.isEmpty() && temperatures[i]>temperatures[s.peek()]){
                 int t = s.pop();
                 answer[t] = i-t;  
             }
                 s.push(i);
             
        }

        return answer;

    }
}

496.下一个更大元素 I

思路

栈头到栈底从小到大
对nums2的元素来进行单调栈的操作，即记录nums2的下标。
但nums2的元素在nums1中出现过的时候，把该索引的位置的元素设置为当前出栈的nums2的元素，即为nums1在nums2中的下一个更大的元素
因为题目强调元素不重复，可以对nums1的元素用map记录，key为元素值，value为索引值，方便用nums2的元素查找1中的索引位置。

代码

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

        return res;
    }
}