一、题目

请根据每日 气温 列表 temperatures ,请计算在每一天需要等几天才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 0 来代替。

点击查看原题
难度级别: 中等

二、思路

1)单调栈

使用暴力法求解思路是很明确的,双重循环,每遍历到一个元素,就向后找到第一次比它大的数字,找不到就为0
思考一下,可不可以用一种数据结构存储后面元素的递增序列,这样就可以直接找到后面元素出现刚好比当前元素的较大值,那就是单调栈,单调栈递减栈的特点:

  • 如果当前元素比栈顶元素大,则出栈,直到栈顶元素比当前元素大

由于从后往前遍历,如果出现了更大的元素值,会截断后面比当前元素小的所有值

三、代码

1)单调栈

  1. class Solution {
  2. public int[] dailyTemperatures(int[] temperatures) {
  3. Stack<Integer> stack = new Stack();
  4. int[] ans = new int[temperatures.length];
  5. for (int i = temperatures.length - 1; i >= 0; i--) {
  6. while (!stack.isEmpty() && temperatures[stack.peek()] <= temperatures[i]) {
  7. stack.pop();
  8. }
  9. ans[i] = stack.isEmpty() ? 0 : (stack.peek() - i);
  10. stack.push(i);
  11. }
  12. return ans;
  13. }
  14. }

时间复杂度为O(n),空间复杂度为O(n)