Leetcode 739. Daily Temperatures

Leetcode 739. Daily Temperatures

算法描述

给一个数组temperatures表示每天的气温，判断多少天后气温高于当前值，输出一个序列intervals表示间隔的天数。

输入: temperatures = [73, 74, 75, 71, 69, 72, 76, 73]
输出: intervals = [1, 1, 4, 2, 1, 1, 0, 0]
题目来源: https://leetcode.com/problems/daily-temperatures/description/

算法思路

思路1: 首先可以想到的思路是遍历数组，更新当前节点之前的所有节点的interval值。时间复杂度为O(n^2)。这样做出来明显会超时。
思路2: 通过分析可以发现，当前节点i的interval值只与其后续节点的interval值相关。如果当前节点i小于下一个节点j，则intervals[i]=1；否则intervals[i]>=1+intervals[j]。因为，根据题意可以发现temperatures[j] < temperatures[1+intervals[j]]，而temperatures[i] > temperatures[j]，所以如果有temperatures[1+intervals[j]] > temperatures[i]，则等号成立，否则大于符号成立，需要做下一步判断。这里的终止条件为处理到某个interval值为0的节点，或者找到某个大于temperatures[i]的节点。时间复杂度小于O(n^2)。

代码实现

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n = temperatures.size();
        vector<int> intervals(n,0);
        for(int i=n-2;i>=0;i--){
            if(temperatures[i]<temperatures[i+1]){
                intervals[i]=1;
                continue;
            }
            int index=1;
            while(i+index<n){
                // 遇到intervals为0的节点就不可能找到更大的节点了
                if(intervals[i+index]==0){
                    intervals[i]=0;
                    break;
                }
                // 更新候选位置
                index+=intervals[i+index];
                // 遇到更大的节点就返回
                if(temperatures[i]<temperatures[i+index]){
                    intervals[i]=index;
                    break;
                }
            }
        }
        return intervals;
    }
};