stock_span.cpp

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

void calcSpan(int price[],int n,int *span) {
  stack<int> s;
  for (int i=0;i<n;i++) {
    while (!s.empty() && price[s.top()] <= price[i]) {
      s.pop();
    }
    if (s.empty()) {
      span[i] = i+1;
    } else {
      span[i] = i - s.top();
    }
    s.push(i);
  }
}

int main () {
  int price[] = { 10,60,70,65,80,85 };
  int n = 6;
  int* span = new int[n];
  for (int i=0;i<n;i++) {
    span[i] = 0;
  }
  calcSpan(price,n,span);
  for (int i=0;i<n;i++) {
      cout<<span[i]<<" ";
  }
  return 0;
}