腾讯2020校园招聘-后台(逛街)

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

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

时间限制:C/C++ 2秒,其他语言4秒

空间限制:C/C++ 256M,其他语言512M

示例1

输入例子:

[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栋楼。  

这是一个典型的单调栈问题。我们可以使用一个单调递减的栈来解决这个问题。遍历每栋楼,如果当前楼的高度大于等于栈顶楼的高度,那么我们就可以确定栈顶楼被当前楼挡住了,所以我们可以将栈顶楼出栈,直到栈为空或者栈顶楼的高度小于当前楼的高度。然后将当前楼入栈。这样,栈中的元素就是当前楼可以看到的楼的数量。

代码实现如下:

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

vector<int> visibleBuildings(vector<int>& buildings) {
    stack<int> stk;
    vector<int> res(buildings.size(), 0);
    
    for (int i = 0; i < buildings.size(); i++) {
        while (!stk.empty() && buildings[i] >= buildings[stk.top()]) {
            stk.pop();
        }
        res[i] = stk.empty() ? i + 1 : i - stk.top();
        stk.push(i);
    }
    
    return res;
}

int main() {
    vector<int> buildings = {5, 3, 8, 3, 2, 5};
    vector<int> res = visibleBuildings(buildings);
    for (int i = 0; i < res.size(); i++) {
        cout << res[i] << " ";
    }
    cout << endl;
    return 0;
}

 

这段代码首先定义了一个单调栈和一个结果数组。然后遍历每栋楼,如果当前楼的高度大于等于栈顶楼的高度,就将栈顶楼出栈,直到栈为空或者栈顶楼的高度小于当前楼的高度。然后将当前楼入栈。最后将栈中的元素作为当前楼可以看到的楼的数量。

  • 11
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

清贫码农

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值