739.每日温度

暴力

代码

//1100ms
public int[] dailyTemperatures(int[] T) {
    if (T == null || T.length == 0) return new int[0];
    int[] ans = new int[T.length];
    for (int i = 0; i < T.length; i++) {
        for (int j = i + 1; j < T.length; j++) {
            if (T[j] > T[i]) {
                ans[i] = j - i;
                break;
            }
        }
    }
    return ans;
}

---

单调栈

原文

图解

代码

//单调栈  16ms
public int[] dailyTemperatures2(int[] T) {
    if (T == null || T.length == 0) return new int[0];
    int[] ans = new int[T.length];
    int len = T.length;
    Deque<Integer> stack = new LinkedList<>();
    for (int i = 0; i < len; i++) {
        int t = T[i];
        while (!stack.isEmpty() && t > T[stack.peek()]) {
            int prevIndex = stack.pop();
            ans[prevIndex] = i - prevIndex;
        }
        stack.push(i);
    }
    return ans;
}

---

动态规划

代码

/**4ms
 * 根据题意，从最后一天推到第一天，这样会简单很多。因为最后一天显然不会再有升高的可能，结果直接为0。
 * 再看倒数第二天的温度，如果比倒数第一天低，那么答案显然为1，如果比倒数第一天高，又因为倒数第一天
 * 对应的结果为0，即表示之后不会再升高，所以倒数第二天的结果也应该为0。
 * 自此我们容易观察出规律，要求出第i天对应的结果，只需要知道第i+1天对应的结果就可以：
 * - 若T[i] < T[i+1]，那么res[i]=1；
 * - 若T[i] > T[i+1]
 *   - res[i+1]=0，那么res[i]=0;
 *   - res[i+1]!=0，那就比较T[i]和T[i+1+res[i+1]]（即将第i天的温度与比第i+1天大的那天的温度进行比较）
 */
public int[] dailyTemperatures(int[] T) {
    int[] res = new int[T.length];
    res[T.length - 1] = 0;
    for (int i = T.length - 2; i >= 0; i--) {
        for (int j = i + 1; j < T.length; j += res[j]) {
            if (T[i] < T[j]) {
                res[i] = j - i;
                break;
            } else if (res[j] == 0) {
                res[i] = 0;
                break;
            }
        }
    }
    return res;
}

//作者： Angus-Liu

参考链接

官方题解：单调栈

数据结构与算法：图文详解