739. Daily Temperatures(python+cpp)

本文介绍了一种使用栈数据结构预测未来温暖天气的算法。通过维护一个降序栈,算法能高效地计算出每日温度直到气温上升所需的天数。适用于气温变化预测场景。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目:

Given a list of daily temperatures T, return a list such that, for each day in the input, tells you how many days you would have to wait until a warmer temperature. If there is no future day for which this is possible, put 0 instead.
For example, given the list of temperatures T = [73, 74, 75, 71, 69, 72, 76, 73], your output should be [1, 1, 4, 2, 1, 1, 0, 0].
Note: The length of temperatures will be in the range [1, 30000]. Each temperature will be an integer in the range [30, 100].

解释:
用栈做,维护一个降序的栈,栈中存储的是indexindex本身不是降序的,但是index对应的T[index]是降序的,当出现比栈中一些index对应的T[index]大的T[i]的时候,把那些index出栈,并保存好位置差,然后把当前的i压入栈,这样就可以实现要求。
python代码:

class Solution:
    def dailyTemperatures(self, T):
        """
        :type T: List[int]
        :rtype: List[int]
        """
        ans=[0]*len(T)
        stack=[]
        for i in range(len(T)):
            while stack and T[stack[-1]]<T[i]:
                cur=stack.pop()
                ans[cur]=i-cur
            stack.append(i)
        return ans

c++代码:

#include <stack>
using namespace std;
class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& T) {
        stack<int>_stack;
        int n=T.size();
        vector<int>result(n,0);
        for(int i=0;i<n;i++)
        {
            while(!_stack.empty() && T[_stack.top()]<T[i])
            {
                int cur=_stack.top();
                _stack.pop();
                result[cur]=i-cur;
            }
            _stack.push(i);
        }
        return result;
    }
};

总结:
while的时候可以多加一些判断。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值