逛街_腾讯笔试

逛街

小Q在周末的时候和他的小伙伴来到大城市逛街，一条步行街上有很多高楼，共有n座高楼排成一行。
小Q从第一栋一直走到了最后一栋，小Q从来都没有见到这么多的楼，所以他想知道他在每栋楼的位置处能看到多少栋楼呢？（当前面的楼的高度大于等于后面的楼时，后面的楼将被挡住）

输入描述:
输入第一行将包含一个数字n，代表楼的栋数，接下来的一行将包含n个数字wi(1<=i<=n)，代表每一栋楼的高度。
1<=n<=100000;
1<=wi<=100000;

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

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

思路: 当Q站在x位置时, 求出从1 到 x 的单调递减栈即是小Q向左看时看到的楼数目, 同理, 求出从n到x从右向左单调递减栈的大小即是在x位置向右看的楼数, 两者相加即可

#include<iostream>
#include<stack>
#include<vector>
#include<algorithm>
using namespace std;
 
int num[100005];
 
int main() {
    ios::sync_with_stdio(false);
    int n;
    cin >> n;
    int j = 0;
    for (int i = 0; i < n; i++)
        cin >> num[j++];
 
    stack<int> s1, s2;
    vector<int> v1, v2;
    for (int i = 0, j = n - 1; i < n, j >= 0; i++, j--) {
        v1.push_back(s1.size());
        v2.push_back(s2.size());
 
        while (!s1.empty() && s1.top() <= num[i]) s1.pop();
        while (!s2.empty() && s2.top() <= num[j]) s2.pop();
 
        s1.push(num[i]);
        s2.push(num[j]);
    }
    reverse(v2.begin(), v2.end());
 
    for (int i = 0; i < n; i++) cout << v1[i] + v2[i] + 1 << " ";
    return 0;
}