Leets 739每日温度

1.题目

2.解法

import java.util.Scanner;
import java.util.Stack;
import java.util.stream.IntStream;

// 求的是当前元素的最小值
public class Main {
	
    public static void main(String[] args) {
    	Scanner in = new Scanner(System.in);
    	String[] strs = in.nextLine().split(" ");
    	
    	int[] tmp = new int[strs.length];
    	for (int i = 0; i < strs.length; i++) {
    		tmp[i] = Integer.parseInt(strs[i]);
    	}
    	
    	Stack<Integer> stack = new Stack<Integer>();
    	int[] days = new int[strs.length];
    	
    	for (int i = 0; i < strs.length; i++) {
    		// 比的是下标的值
    		while(!stack.isEmpty() && tmp[i] > tmp[stack.peek()]) {
    			int d = stack.pop();
    			days[d] = i - d;
    		}
    		// 存的是下标
    		stack.push(i);
    	}
    	
    	for(int d : days) {
    		System.out.print(d + " ");
    	}
    }
}