-
Notifications
You must be signed in to change notification settings - Fork 46
/
Stock_span_problem.cpp
68 lines (59 loc) · 1.66 KB
/
Stock_span_problem.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#include<bits/stdc++.h>
using namespace std;
/*problem statement:-
The stock span problem is a financial problem where we have a series of n daily price quotes for a stock and we need to calculate the span of stocks price for all n days.
The span Si of the stocks price on a given day i is defined as the maximum number of consecutive days just before the given day, for which the price of the stock on the current day is less than or equal to its price on the given day.
For example, if an array of 7 days prices is given as {100, 80, 60, 70, 60, 75, 85}, then the span values for corresponding 7 days are {1, 1, 1, 2, 1, 4, 6}.
Input:
N = 7, price[] = [100 80 60 70 60 75 85]
Output:
1 1 1 2 1 4 6
*/
class Solution
{
public:
//Function to calculate the span of stock’s price for all n days.
vector <int> calculateSpan(int price[], int n)
{
vector<int> v;
stack<pair<int,int>> s;
v.push_back(1);
s.push({price[0],1});
for(int i=1;i<n;i++){
int count=0;
while(s.size()!=0){
auto it=s.top();
if(it.first>price[i]){
break;
}
else{
count=count+it.second;
s.pop();
}
}
v.push_back(count+1);
s.push({price[i],count+1});
}
return v;
}
};
int main(){
// taking input:
int t;
cin>>t;
while(t--){
int n;
cin>>n;
int i,a[n];
for(i=0;i<n;i++){
cin>>a[i];
}
Solution obj;
vector<int> s=obj.calculateSpan(a,n);
for(i=0;i<n;i++){
cout<<s[i]<<" ";
}
cout<<endl;
}
return 0;
}