目录
问题描述
小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栋楼。
思路
想法1
最直接的想法,针对每一栋楼,分别向前向后记录可以看到的楼层数,时间复杂度为O(n*n)。程序运行时间会超时,所以还不够。
想法2
利用【单调栈】的思想,针对5 3 8 3 2 5
i | (0->n-1)楼数量front[i] | 单调栈 | i | (n-1->0)楼数量back[i] | 单调栈 | i | front[i]+back[i]+1 | ||
0 | 0 | 5 | 6 | 0 | 5 | 0 | 3 | ||
1 | 1 | 5,3 | 5 | 1 | 5,2 | 1 | 3 | ||
2 | 2 | 8 | 4 | 2 | 5,3 | 2 | 5 | ||
3 | 1 | 8,3 | 3 | 2 | 8 | 3 | 4 | ||
4 | 2 | 8,3,2 | 2 | 1 | 8,3 | 4 | 4 | ||
5 | 3 | 8,5 | 1 | 2 | 8,5 | 5 | 4 | ||
6 | 2 | \ | 0 | 2 | \ | 6 | 4 |
从表格中看,位置0可以看到的楼层数为0+2+1。也就是向前看没有楼层,向后能看到位置1和位置2的楼层,位置2楼高8,后面的无法看到。再加上能看到位置0的楼。因此最终能看到的楼的数量为3。
代码
#include<iostream>
#include<stack>
using namespace std;
int main()
{
int n;
cin>>n;
int* build=new int[n];
int* num=new int[n];
for(int i=0;i<n;i++)
{
cin>>build[i];
num[i]=1;
}
stack<int> q;
for(int i=0;i<n;i++)
{
num[i]+=q.size();
while(!q.empty()&&q.top()<=build[i])
{
q.pop();
}
q.push(build[i]);
}
stack<int> p;
for(int i=n-1;i>=0;i--)
{
num[i]+=p.size();
while(!p.empty()&&p.top()<=build[i])
{
p.pop();
}
p.push(build[i]);
}
for(int i=0;i<n;i++)
{
cout<<num[i];
if(i!=n-1)
cout<<" ";
}
return 0;
}