Leecode-739 每日温度

目录

      • 题目要求:
      • 思路1:
        • 代码:
      • 思路2:
        • 代码:

题目要求:

根据每日 气温 列表,请重新生成一个列表,对应位置的输出是需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。

例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。

思路1:

第一次看到这个题的时候,想着先从简单的办法写起:
那么就来了,就是外层for循环 i 遍历数组,内层循环 从 j = i + 1开始遍历,找寻大于外层循环元素的值,

  • 找到了 结果数组 res[i] = j - i
  • 到了数组末尾了还没找到,res[i] = 0
代码:
public static int[] dailyTemperaturesDemo(int[] T){
        int len = T.length;
        int[] res = new int[len];

        for(int i = 0; i < len;i++) {
            int j = i + 1;
            while (j < len && T[j] <= T[i]){
                j++;
            }
            if(j == len)
                res[i] = 0;
            else
                res[i] = j - i;
        }
        return res;
    } 

但是后面想着用别的办法解的时候,却迟迟想不出来,所以看了题解,发现正确用对数据结构,真的很重要。

思路2:

很明显,可以看出,前面的方法,有多次重复计算的过程。
比如数组: {73,74,75,71,69,72,76,73} 当遍历到75时, 后面几个数都是小于它的,内层循环肯定会遍历,直到找到76这个数,内层循环才会结束。 然后外层循环遍历到71, 它的内层循环又会重复计算69与71的大小关系。
所以优化就是为了找到避免重复计算的方法。

递减栈解法:
遍历整个数组,如果栈不空,且当前数字大于栈顶元素,那么如果直接入栈的话就不是 递减栈 ,所以需要取出栈顶元素,由于当前数字大于栈顶元素的数字,而且一定是第一个大于栈顶元素的数,直接求出下标差就是二者的距离。
继续看新的栈顶元素,直到当前数字小于等于栈顶元素停止,然后将数字入栈,这样就可以一直保持递减栈,且每个数字和第一个大于它的数的距离也可以算出来。

作者:MisterBooo
链接

代码:
	public static int[] dailyTemperaturesDemo2(int[] T) {
        if(T == null || T.length == 0)
            return null;
        int ans[] = new int[T.length];
        Stack<Integer> stack = new Stack<>();
        int len = T.length;
        for(int i = 0; i < len;i++) {
            //遇到比当前栈顶元素大的T[i] 弹出栈顶元素,然后将结果数组ans的值更改,如果T[i] < 当前栈顶元素, 则入栈
            while (!stack.isEmpty() && T[i] > T[stack.peek()]) {
                int top = stack.pop();
                ans[top] = i - top;
            }
            stack.push(i);
        }
        return ans;
    }

此解法,leecode用时18ms,而第一个解法用时925ms 孰强孰弱,一眼便知。 所以说选择一些好的数据结构解题,真的会事半功倍。
加油,路还长~

还有一些类似题:

  • 求解算术表达式的结果(LeetCode 224、227、772、770)
  • 求解直方图里最大的矩形区域(LeetCode 84)

你可能感兴趣的:(数据结构与算法,java,数据结构,leetcode)