Time: 27 ms (89.53%), Space: 55.4 MB (43.25%) - LeetHub

Author: Vivekchavda1374

0937-online-stock-span/0937-online-stock-span.java

@@ -1,24 +1,14 @@
 class StockSpanner {
-    stack<pair<int,int>> s; //1st is index and 2nd is value
-    int index;
-public:
-    StockSpanner() {
-        ios::sync_with_stdio(false);
-        cin.tie(0);
-        index = -1;
+    Stack<int[]> st;
+    public StockSpanner() {
+        st = new Stack<>();
     }
-    
-    int next(int price) {
-        index +=1;
-        
-        while(!s.empty() && s.top().second<=price)    //Find the previous greater element
-            s.pop();
-        //If there is no previous greater element
-        if(s.empty())
-        {   s.push({index,price});      return index+1;   }
-        
-        int result = s.top().first;
-        s.push({index,price});
-        return index-result;        
+    public int next(int price) {
+        int span = 1;
+        while (!st.isEmpty() && st.peek()[0] <= price) {
+            span += st.pop()[1];
+        }
+        st.push(new int[] {price, span});
+        return span;
     }
-};
+}