Stock Span Problem Solution

stock_span.java

import java.util.*;

class ValueWithCount {
    int value;
    int index;

    ValueWithCount(int value, int index) {
        this.value = value;
        this.index = index;
    }
}

class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        Stack<ValueWithCount> stack = new Stack<>();
        int[] arr = new int[N];

        for(int i=0; i<N; i++) {
            arr[i] = sc.nextInt();
            int count = 1;
            while(stack.size() !=0 && stack.peek().value <= arr[i]) {
                count += stack.pop().index;
            }
            stack.push(new ValueWithCount(arr[i], count));
            System.out.print(count +" ");
        }

    }
}