腾讯校园招聘笔试——逛街能看到楼的数量

小Q在周末的时候和他的小伙伴来到大城市逛街，一条步行街上有很多高楼，共有n座高楼排成一行。

小Q从第一栋一直走到了最后一栋，小Q从来没有看到过这么多高楼，所以他想知道他在每栋楼的位置处能看到多少栋楼呢？（当前面的楼的高度大于等于后面的楼时，后面的楼将被挡住）

输入描述：

输入第一行将包含一个数字n,表示楼的栋数，接下来的一行将包含n个数字wi(1<=i<=n)，代表一栋楼的高度。

1<=n<=100000;

1<=wi<=100000;

输出描述：

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

单调栈法

#include <iostream>
#include <stack>
using namespace std;

int main()
{
    int n;
    cin>>n;
    int height[n];
    for(int i=0;i<n;i++)
    {
        cin>>height[i];
    }
    int right[n];
    int left[n];
    stack<int> rightStack;
    stack<int>leftStack;
    //从左往右看
    for(int i=n-1;i>=0;i--)
    {
        right[i]=rightStack.size();
        if(i==n-1)
        {
            rightStack.push(height[i]);
        }
        else if(height[i]<rightStack.top())
        {
            rightStack.push(height[i]);
        }
        else
        {
            while(rightStack.size()!=0&&height[i]>=rightStack.top())
            {
                rightStack.pop();
            }
            rightStack.push(height[i]);
        }
    }
    //从右往左看
    for(int i=0;i<n;i++)
    {
        left[i]=leftStack.size();
        if(i==0)
        {
            leftStack.push(height[i]);
        }
        else if(height[i]<leftStack.top())
        {
            leftStack.push(height[i]);
        }
        else
        {
            while(leftStack.size()!=0&&height[i]>=leftStack.top())
            {
                leftStack.pop();
            }
            leftStack.push(height[i]);
        }
    }
    for(int i=0;i<n;i++)
    {
        cout<<right[i]+left[i]+1<<" ";
    }
	return 0;
}