单调栈

腾讯2020后端笔试题 逛街

 

#include<bits/stdc++.h>

using namespace std;

int main(){
    int n;
    while(cin >> n){
        vector<int> s(n);
        for(int i=0;i<n;i++)
            cin >> s[i];
        stack<int> sta;
        vector<int> left(n);
        vector<int> right(n);
        for(int i=0;i<n;i++){
            left[i] = sta.size();
            while(!sta.empty()&&sta.top()<=s[i])
                sta.pop();
            sta.push(s[i]);    
        }
        while(!sta.empty())
            sta.pop();
        for(int j=n-1;j>=0;j--){
            right[j] = sta.size();
            while(!sta.empty()&&sta.top()<=s[j])
                sta.pop();
            sta.push(s[j]);
        }
        for(int i=0;i<n;i++)
            cout << left[i]+1+right[i] <<" ";
    }
    
    return 0;
}