LeetCode : 739. 每日温度

739. 每日温度

题目描述 提示帮助 提交记录 社区讨论 阅读解答

即将到来

随机一题

---

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

例如，给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]，你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示：气温 列表长度的范围是 [1, 30000]。每个气温的值的都是 [30, 100] 范围内的整数。

/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* dailyTemperatures(int* temperatures, int temperaturesSize, int* returnSize) {
    struct my_stack
    {
        int num;
        int start;
    };
    int* ans = (int *)malloc(sizeof(int) * temperaturesSize);
    struct my_stack* stack = (struct my_stack*)malloc(sizeof(struct my_stack) * temperaturesSize);
    int flag = 0;
    
    for(int i = 0;i < temperaturesSize;i++)
        ans[i] = 0;
    
    stack[flag].num = temperatures[0];
    stack[flag++].start = 0;
    for(int i = 1;i < temperaturesSize;i++){
        if(flag == 0 || temperatures[i] <= stack[flag - 1].num){
            stack[flag].num = temperatures[i];
            stack[flag++].start = i;
        }
        else if(temperatures[i] > stack[flag - 1].num){
            while(stack[flag - 1].num < temperatures[i]){
                if(flag > 0 && stack[flag - 1].start >= 0)
                    ans[stack[flag - 1].start] = i - stack[flag - 1].start;
                else
                    break;
                flag--;
            }
            stack[flag].num = temperatures[i];
            stack[flag++].start = i;
        }
    }
    *returnSize = temperaturesSize;
    return ans;
    
}

这题的时间复杂度是O(n)。主要的思想是通过模拟栈的方式来实现的，首先是判断如果是该栈为空或者temperatures[i]小于栈顶元素的时候进行压栈操作，因为当该元素小于栈顶元素的时候不需要进行任何操作，而当出现 temperatures[i]大于栈顶元素的时候则进行比较，如果栈中元素小于temperatures[i]时则进行出栈操作直到栈空，然后将此时的temperatures[i]压栈然后重复之前的操作。注意这题重点在于当后一个数小于前一个数的时候是没有任何意义的，只有当出现更大的数时才进行操作。