Time: 140 ms (83.34%), Space: 85.1 MB (70.13%) - LeetHub

arpitkekri · Nov 13, 2021 · 0706ab9 · 0706ab9
1 parent 42fe79b
commit 0706ab9
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/daily-temperatures/daily-temperatures.cpp b/daily-temperatures/daily-temperatures.cpp
@@ -0,0 +1,20 @@
+class Solution {
+public:
+    vector<int> dailyTemperatures(vector<int>& temperatures) {
+        stack<pair<int, int>> stk;
+        int n = temperatures.size();
+        vector<int> waitDays(n, 0);
+        for(int i = n-1; i >= 0; i--) {
+            if(stk.empty()) 
+                stk.push({temperatures[i], i});
+            else {
+                while(!stk.empty() && stk.top().first <= temperatures[i])
+                    stk.pop();
+                if(!stk.empty())
+                    waitDays[i] = stk.top().second - i;
+                stk.push({temperatures[i], i});
+            }
+        }
+        return waitDays;
+    }
+};