LeetCode739 每日温度

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

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

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

 


 

 

//章节 - 队列和栈    
//三、栈:先入后出的数据结构
//3.每日温度
/*
算法思想:
    审请题意,意思是找一个比当前温度高的最近天数。使用递减栈Descending Stack来做,栈里只有递减元素,思路是这样的,我们遍历数组,如果栈不空,且当前数字大于栈顶元素,那么如果直接入栈的话就不是递减栈了,所以我们取出栈顶元素,那么由于当前数字大于栈顶元素的数字,而且一定是第一个大于栈顶元素的数,那么我们直接求出下标差就是二者的距离了,然后继续看新的栈顶元素,直到当前数字小于等于栈顶元素停止,然后将数字入栈,这样就可以一直保持递减栈,且每个数字和第一个大于它的数的距离也可以算出来了
*/
//算法实现:
class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n = temperatures.size();
        vector<int> res(n, 0);
        stack<int> st;
        for (int i = 0; i < n; ++i) {
            while (!st.empty() && temperatures[i] > temperatures[st.top()]) {
                int t = st.top();   //返回栈顶元素,注意这是返回的是下标
                st.pop();
                res[t] = i - t; //下标差即为距离
            }
            st.push(i); //注意,这是是入栈下标
        }
        return res;
    }
};

 

全部评论
import copy if __name__ == '__main__': temperatures = [73,74,75,71,69,72,76,73] # temperatures = [30, 60, 90] # temperatures = [30, 40, 50, 60] later_temperatures = copy.copy(temperatures) day_nums = [] for temperature in temperatures: later_temperatures.remove(later_temperatures[0]) i = 1 day_num = 0 for later_temperature in later_temperatures: if later_temperature > temperature: day_num = i i = i + 1 break else: i = i +1 if i == 1: day_num = 0 day_nums.append(day_num) print(day_nums)
点赞
送花
回复
分享
发布于 2021-11-20 17:42

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务