腾讯笔试题之【逛街】

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

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

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

int main()
{
    int n;
    cin >> n;
    vector<int> arr(n);
    vector<int> dp(n);//保存从前往后和从后往前遍历的结果
    for (int i = 0; i < n; i++)
        cin >> arr[i];
    
    stack<int> l,r;//单调栈
    
    //从后往前遍历，找到每个位置后面能看到的房子个数
    for(int i = n-1;i > 0;i--)
    {
    	//如果栈顶元素比当前元素小，说明我这个高个子挡住后面矮个子的房子了
    	//那么从当前位置往后面看，肯定就看不到矮个子的房子，只看到高个子的房子
    	//所以可以直接pop掉矮个子的房子
        while (!r.empty() && arr[i] >= r.top())
            r.pop();
        r.push(arr[i]);
        dp[i - 1] = r.size();
        cout << dp[i - 1] << " ";
    }
    cout << endl;

	//注意边界问题：从后往前没有计算到0，在1的时候就把0计算了
    cout<<dp[0] + 1<<" ";
    //从后往前遍历，找到每个位置前面能看到的房子个数
    for(int i = 0;i < n-1;i++)
    {
    	//同理
        while (!l.empty() && arr[i] >= l.top())
            l.pop();
        l.push(arr[i]);
        cout << l.size() << "++";
        dp[i + 1] += l.size() + 1;
        cout<<dp[i + 1]<<" ";
    }
}