hihocoder - 股票价格3 - 维护栈 & 线段树

题目

时间限制:10000ms
单点时限:1000ms
内存限制:256MB
描述
小Hi最近在关注股票，为了计算股票可能的盈利，他获取了一只股票最近N天的价格A1~AN。

小Hi想知道，对于第i天的股票价格Ai，几天之后股价会第一次超过Ai。

假设A=[69, 73, 68, 81, 82]，则对于A1=69，1天之后的股票价格就超过了A1；对于A2=73，则是2天之后股票价格才超过A2。

输入
第一行包含一个整数N。

以下N行每行包含一个整数Ai。

对于50%的数据，1 ≤ N ≤ 1000

对于100%的数据，1 ≤ N ≤ 100000, 1 ≤ Ai ≤ 1000000

输出
输出N行，其中第i行代表对于第i天的股票价格Ai，几天之后股价会第一次超过Ai。

如果Ai+1~AN之内没有超过Ai，输出-1。

样例输入
5
69
73
68
81
82
样例输出
1
2
1
1
-1

题解

单调栈法

#include<stdio.h>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<map>
#include<vector>
#include<queue>
#include<stack>
#include<deque>
using namespace std;
typedef long long LL;
#define ms(a,b) memset(a,b,sizeof(a))
#define eps 1e-8

const int maxn=1e5+100;
int A[maxn];
int B[maxn];

int main() {
    int n;
    cin>>n;
    for(int i=0;i<n;i++) cin>>A[i];
    stack<pair<int,int> >s;
    for(int i=0;i<n;i++){
        if(s.empty() || s.top().first>=A[i]){
            s.push({A[i],i});
        }else{
            while(!s.empty() && s.top().first<A[i]){
                B[s.top().second]=i-s.top().second;
                s.pop();
            }
            s.push({A[i],i});
        }
    }
    while (!s.empty()) {
        B[s.top().second] = -1;
        s.pop();
    }
    for(int i=0;i<n;i++) printf("%d\n",B[i]);
    return 0;
}