小Q在周末的时候和他的小伙伴来到大城市逛街,一条步行街上有很多高楼,共有n座高楼排成一行。
小Q从第一栋一直走到了最后一栋,小Q从来都没有见到这么多的楼,所以他想知道他在每栋楼的位置处能看到多少栋楼呢?(当前面的楼的高度大于等于后面的楼时,后面的楼将被挡住)
输入描述:
输入第一行将包含一个数字n,代表楼的栋数,接下来的一行将包含n个数字wi(1<=i<=n),代表每一栋楼的高度。
1<=n<=100000;
1<=wi<=100000;
输出描述:
输出一行,包含空格分割的n个数字vi,分别代表小Q在第i栋楼时能看到的楼的数量。
输入例子1:
6
5 3 8 3 2 5
输出例子1:
3 3 5 4 4 4
例子说明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;
}