Update 496.Next Greater Element I.cpp

wisdompeak · web-flow · commit fd8942430654 · 2021-12-17T06:27:07.000-08:00
diff --git a/Stack/496.Next-Greater-Element-I/496.Next Greater Element I.cpp b/Stack/496.Next-Greater-Element-I/496.Next Greater Element I.cpp
@@ -7,17 +7,12 @@ class Solution {
         
         for (int i=0; i<nums.size(); i++)
         {
-            if (Stack.empty() || Stack.top()>nums[i])
-                Stack.push(nums[i]);
-            else
+            while (!Stack.empty() && Stack.top()<nums[i])
             {
-                while (!Stack.empty() && Stack.top()<nums[i])
-                {
-                    Map[Stack.top()]=nums[i];
-                    Stack.pop();
-                }
-                Stack.push(nums[i]);
-            }
+                Map[Stack.top()]=nums[i];
+                Stack.pop();
+            }                            
+            Stack.push(nums[i]);
         }
         
         vector<int>results;
