每日温度

给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 res ,其中 res[i]是指在第 i 天之后,才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 0 来代替。

暴力解法:两层for循环即可

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int m = temperatures.length;
        int[] answer = new int[m];
        for (int i = 0; i < m; i++) {
            for (int j = i + 1; j < m; j++) {
                if (temperatures[i] < temperatures[j]) {
                    answer[i] = j - i;
                    break;
                }
            }
        }
        return answer;
    }
}

解法二:单调栈
通常是一维数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置,此时就要想到可以用单调栈了

例如本题其实就是找到一个元素右边第一个比自己大的元素。

使用单调栈的时候首先要明确如下几点:

  • 单调栈里只需要存放元素的下标i就可以了,如果需要使用对应的元素,直接T[i]就可以获取
  • 单调栈里元素是递增呢? 还是递减呢?

使用单调栈主要有三个判断条件:

  • 当前遍历的元素temperatures[i]小于栈顶元素temperatures[stack.peek()]的情况
  • 当前遍历的元素temperatures[i]等于栈顶元素temperatures[stack.peek()]的情况
  • 当前遍历的元素temperatures[i]大于栈顶元素temperatures[stack.peek()]的情况
class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int len = temperatures.length;
        int[] res = new int[len];
        Stack<Integer> stack = new Stack<>();
        stack.push(0);
        for (int i = 1; i < len; i++) {
            //当前遍历的元素temperatures[i]小于或等于栈顶元素temperatures[st.peek()]
            if (temperatures[i] <= temperatures[stack.peek()]) {
                stack.push(i);
            } else {
                //当前遍历的元素temperatures[i]大于栈顶元素temperatures[stack.peek()]
                while (!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]) {
                    res[stack.peek()] = i - stack.pop();
                }
                stack.push(i);
            }
        }
        return res;
    }
}
posted @ 2022-05-20 11:53  Arthurma  阅读(76)  评论(0编辑  收藏  举报