LeetCode/NeetCodeRoadmap/Stack/739_Daily_Temperatures.py

46 lines
1.5 KiB
Python

class Solution(object):
def dailyTemperatures(self, temperatures):
"""
:type temperatures: List[int]
:rtype: List[int]
"""
count = 0
warmer_days = []
l = 0
r = 1
while l < r:
# print(r)
if r == len(temperatures)-1:
if l != len(temperatures)-2:
if temperatures[r] > temperatures[l]:
count += 1
warmer_days.append(count)
else:
warmer_days.append(0)
l += 1
r = l + 1
count = 0
else:
if temperatures[r] > temperatures[l]:
count += 1
warmer_days.append(count)
else:
warmer_days.append(0)
break
elif temperatures[r] > temperatures[l]:
count += 1
warmer_days.append(count)
count = 0
l += 1
r = l + 1
else:
r += 1
count += 1
warmer_days.append(0)
return warmer_days
obj = Solution()
# print(obj.dailyTemperatures([73,74,75,71,69,72,76,73])) # [1,1,4,2,1,1,0,0]
# print(obj.dailyTemperatures([30,40,50,60])) # [1,1,1,0]
# print(obj.dailyTemperatures([55,38,53,81,61,93,97,32,43,78])) # [3,1,1,2,1,1,0,1,1,0]
print(obj.dailyTemperatures([77,77,77,77,77,41,77,41,41,77])) # [0,0,0,0,0,1,0,2,1,0]