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