Update 0739-每日温度.py

Author: zaie

0739.每日温度/0739-每日温度.py

@@ -1,15 +1,18 @@
-class Solution(object):
-    def dailyTemperatures(self, T):
-        """
-        :type T: List[int]
-        :rtype: List[int]
-        """
-        res = [0] * len(T)
-        s = []
-        # print res
-        for i in range(0, len(T)):
-            while(s and T[i] > T[s[-1]]):
-                res[s[-1]] = i - s[-1]
-                s.pop()
-            s.append(i)            
-        return res
+# 单调栈解法
+class Solution:
+    def dailyTemperatures(self, T: List[int]) -> List[int]:
+        # 可以维护一个存储下标的单调栈，从栈底到栈顶的下标对应的温度列表中的温度依次递减。
+        # 如果一个下标在单调栈里，则表示尚未找到下一次温度更高的下标。
+        ans = [0] * len(T)
+        stack = []
+        for i in range(len(T)):
+            while stack and T[stack[-1]] < T[i]:    # 栈不为空 && 栈顶温度小于当前温度
+                ans[stack[-1]] = i - stack[-1]
+                stack.pop()
+            stack.append(i)
+        return ans
+
+# 作者：dz-lee
+# 链接：https://leetcode.cn/problems/daily-temperatures/solutions/284113/dan-diao-zhan-cong-hou-xiang-qian-bian-li-python3x/
+# 来源：力扣（LeetCode）
+# 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。