maxArea.c++

#include <iostream>
#include <stack>
#include <vector>
#include <climits>
using namespace std;

class Solution {
public:
    vector<int> nextSmallerElement(vector<int> arr, int n) {
        stack<int> s;
        s.push(-1);
        vector<int> ans(n);

        for(int i=n-1; i>=0 ; i--) {
            int curr = arr[i];
            while(s.top() != -1 && arr[s.top()] >= curr)
            {
                s.pop();
            }
            //ans is stack ka top
            ans[i] = s.top();
            s.push(i);
        }
        return ans;
    }
    
    vector<int> prevSmallerElement(vector<int> arr, int n) {
        stack<int> s;
        s.push(-1);
        vector<int> ans(n);

        for(int i=0; i<n; i++) {
            int curr = arr[i];
            while(s.top() != -1 && arr[s.top()] >= curr)
            {
                s.pop();
            }
            //ans is stack ka top
            ans[i] = s.top();
            s.push(i);
        }
        return ans; 
    }
    

    int largestRectangleArea(vector<int>& heights)
    {
        int n= heights.size();
        
        vector<int> next(n);
        next = nextSmallerElement(heights, n);
            
        vector<int> prev(n);
        prev = prevSmallerElement(heights, n);
        
        int area = INT_MIN;
        for(int i=0; i<n; i++)
        {
            int l = heights[i];
            
            if(next[i] == -1)
            {
                next[i] = n;
            }
            int b = next[i] - prev[i] - 1;
            int newArea = l*b;
            area = max(area, newArea);
        }
        return area;
    }
};

int main()
{
    vector<int> heights;
    heights.push_back(2);
    heights.push_back(4);
    heights.push_back(6);
    heights.push_back(3);
    heights.push_back(1);

    Solution s;
    int ans = s.largestRectangleArea(heights);
    vector<int> nextSmaller = s.nextSmallerElement(heights, 5);
    vector<int> prevSmaller = s.prevSmallerElement(heights, 5);

    for(int i = 0; i < 5; i++)
    {
        cout << nextSmaller[i] << " ";
    } cout << endl;
    for(int i = 0; i < 5; i++)
    {
        cout << prevSmaller[i] << " ";
    } cout << endl;
    for(int i = 0; i < 5; i++)
    {
        cout << heights[i] << " ";
    } cout << endl;

    cout << ans << endl;
}