代码随想录算法刷题训练营day48： LeetCode(739) 每日温度、LeetCode(496)下一个更大元素 I 、LeetCode(503)下一个更大元素II

代码随想录算法刷题训练营day48： LeetCode(739) 每日温度、LeetCode(496)下一个更大元素 I 、LeetCode(503)下一个更大元素II

LeetCode(739) 每日温度
题目

代码

import java.util.Stack;

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int[] result=new int[temperatures.length];//创建一个数组记录数据
        Stack<Integer> stack=new Stack<Integer>();
        //通过数组和单调栈结合出结果
        for(int i=0;i<temperatures.length;i++){
            if(stack.isEmpty()){
                stack.push(i);//直接将数组下标放入
            }else{
                //比较遍历数组和栈口元素的大小
                while(!stack.isEmpty()&&temperatures[i]>temperatures[stack.peek()]){//注意先后顺序
                    result[stack.peek()]=i-stack.peek();
                    stack.pop();
                }
                stack.push(i);
            }
        }
        return result;
    }
}

LeetCode(496)下一个更大元素 I
题目

代码

import java.util.HashMap;
import java.util.Map;
import java.util.Stack;

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        //单调栈+map去做
        int[] result=new int[nums1.length];
        for(int i=0;i<result.length;i++){
            result[i]=-1;
        }
        //定义一个map集合进行映射
        Map<Integer,Integer> map=new HashMap<>();
        Stack<Integer> stack=new Stack<>();
        for(int i=0;i<nums2.length;i++){
            if(stack.isEmpty()){
                stack.push(nums2[i]);
            }else{
                while(!stack.isEmpty()&&nums2[i]>stack.peek()){
                    map.put(stack.peek(), nums2[i]);
                    stack.pop();
                }
                stack.push(nums2[i]);
            }
        }
        for(int i=0;i<nums1.length;i++){
            if(map.containsKey(nums1[i])){
                result[i]=map.get(nums1[i]);
            }  
        }
        return result;

    }
}

LeetCode(503)下一个更大元素II
题目

代码

import java.util.Stack;

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        //循环数组，拼一下
        int[] date=new int[nums.length*2];
        for(int i=0;i<date.length;i++){
            date[i]=nums[i%nums.length];
        }
        Stack<Integer> stack=new Stack<>();
        int[] resultTemp=new int[date.length];
        for(int i=0;i<resultTemp.length;i++){
            resultTemp[i]=-1;
        }
        for(int i=0;i<date.length;i++){
            if(stack.isEmpty()){
                stack.push(i);
            }else{
                while(!stack.isEmpty()&&date[i]>date[stack.peek()]){
                    resultTemp[stack.peek()]=date[i];
                    stack.pop();
                }
                stack.push(i);
            }
        }
        int[] result=new int[nums.length];
        for(int i=0;i<result.length;i++){
           result[i]=resultTemp[i];
        }
        return result;
    }
}