leetcode-739 每日温度 Python

最新推荐文章于 2023-02-27 17:05:02 发布

原创最新推荐文章于 2023-02-27 17:05:02 发布 · 382 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #数据结构 #算法 #leetcode

本文介绍了三种不同的算法实现方式，用于预测每日气温变化。方法一采用从后往前遍历的策略，方法二利用栈来优化计算过程，而方法三则提出了正向栈的解决方案。这些算法在处理气温预测问题上表现出了高效性和准确性。

部署运行你感兴趣的模型镜像

方法一：从后往前

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
        Next = [float('inf')]*102
        ans = [0] * len(T)
        for i in range(len(T)-1, -1, -1):
            warmer = min(Next[t] for t in range(T[i]+1, 102))
            if warmer < float('inf'):
                ans[i] = warmer - i
            Next[T[i]] = i
        return ans

方法二：栈

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

方法三：栈（正向）

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
        class Node:
            def __init__(self, t, index):
                self.t = t
                self.index = index
        stack = []
        ans = [0] * len(T)
        for i in range(len(T)):
            while stack and T[i] > stack[-1].t:
                top = stack.pop(-1)
                ans[top.index] = i - top.index
            node = Node(T[i], i)
            stack.append(node)
        while stack:
            top = stack.pop(-1)
            ans[top.index] = 0
        return ans