Ir de compras

Durante el fin de semana, Xiao Q y sus amigos vinieron a la gran ciudad para ir de compras. Hay muchos edificios altos en una calle peatonal, y hay n edificios altos alineados en una fila.

Xiao Q caminó desde el primer edificio hasta el último. Xiao Q nunca había visto tantos edificios, por lo que quiere saber cuántos edificios puede ver en la ubicación de cada edificio. (Cuando la altura del edificio delantero es mayor o igual que el edificio trasero, el edificio trasero se bloqueará) 

 

Ingrese descripción:

La primera línea de entrada contendrá un número n, que representa el número de edificios, y la siguiente línea contendrá n números wi (1 <= i <= n), que representan la altura de cada edificio.

1 <= n <= 100000;

1 <= wi <= 100000; 

 

Descripción de salida:

输出一行，包含空格分割的n个数字vi，分别代表小Q在第i栋楼时能看到的楼的数量。

 

Ejemplo de entrada 1:

6
5 3 8 3 2 5

 

Ejemplo de salida 1:

3 3 5 4 4 4

 

Descripción de ejemplo 1:

当小Q处于位置3时，他可以向前看到位置2,1处的楼，向后看到位置4,6处的楼，加上第3栋楼，共可看到5栋楼。当小Q处于位置4时，他可以向前看到位置3处的楼，向后看到位置5,6处的楼，加上第4栋楼，共可看到4栋楼。

 

#include <bits/stdc++.h>
using namespace std;

int main(){
	int N;
	cin>>N;
	vector<int> vec(N);
	for (int i = 0; i <= N; i++){
	    cin >> vec[i];
    }
    stack<int> sta;
	vector<int> look_left(N,0);
	//向左看，不包括所在的楼
	for (int i = 1; i < N; ++i) {
		if (sta.empty() || sta.top() > vec[i-1]) {
			sta.push(vec[i-1]);
		}
		else {
			while (!sta.empty() && sta.top() <= vec[i-1]){
				sta.pop();
			}
			sta.push(vec[i-1]);
		}
		look_left[i]=sta.size();
	}
	//向右看，不包括所在的楼
	vector<int> look_right(N,0);
	sta = stack<int>();
	for (int i= N - 2; i >= 0; --i){
		if (sta.empty() || sta.top() > vec[i+1]) {
			sta.push(vec[i+1]);
		}
		else{
			while(!sta.empty() &&  sta.top() <= vec[i+1]){
				sta.pop();
			}
			sta.push(vec[i+1]);
		}
		look_right[i] =sta.size();
	}
	for (int i=0;i<N;++i){
		cout <<  1 + look_left[i] + look_right[i] << " ";
	}
	cout << endl;
}