Update Stock Span

Author: akshayavb99

Course 2 - Data Structures in JAVA/Lecture 9 - Stacks/Stock Span

@@ -34,3 +34,40 @@ Sample Input 2:
 Sample Output 2:
 1 2 3 4 1 1 2 8 
 */
+import java.util.Stack;
+public class Solution {
+
+	public static int[] stockSpan(int[] price) {
+		//Your code goes here
+        int[] span=new int[price.length];
+        Stack<Integer> stack=new Stack<Integer>();
+        span[0]=1;
+        stack.push(0);
+        
+        for (int i=1;i<price.length;i++)
+        {
+            while(!stack.isEmpty() && price[stack.peek()]<price[i])
+            {
+                stack.pop();
+            }
+            //System.out.println(stack.size());
+            
+            if (stack.isEmpty())
+            {
+                span[i]=i+1;
+            }
+            else
+            {
+                //System.out.println("i = "+i);
+                //System.out.println("top = "+stack.peek());
+                span[i]=i-stack.peek();
+            }
+            stack.push(i);
+            //System.out.println("New top: "+stack.peek());
+            //System.out.println();
+        }
+        return span;
+        
+	}
+
+}