时间限制:C/C++ 2秒,其他语言4秒
空间限制:C/C++ 256M,其他语言512M
小Q在周末的时候和他的小伙伴来到大城市逛街,一条步行街上有很多高楼,共有n
座高楼排成一行。
小Q从第一栋一直走到了最后一栋,小Q从来都没有见到这么多的楼,所以他想知道
他在每栋楼的位置处能看到多少栋楼呢?(当前面的楼的高度大于等于后面的楼时,后
面的楼将被挡住)
输入描述:
输入第一行将包含一个数字n,代表楼的栋数,接下来的一行将包含n个数字wi
(1<=i<=n),代表每一栋楼的高度。
1<=n<=100000;
1<=wi<=100000;
输出描述:
输出一行,包含空格分割的n个数字vi,分别代表小Q在第i栋楼时能看到的楼的数量。
输入例子1:
6
5 3 8 3 2 5
输出例子1:
3 3 5 4 4 4
例子说明1:
当小Q处于位置3时,他可以向前看到位置2,1处的楼,向后看到位置4,6处的楼,加
上第3栋楼,共可看到5栋楼。当小Q处于位置4时,他可以向前看到位置3处的楼,向后看
到位置5,6处的楼,加上第4栋楼,共可看到4栋楼。
刚开始看到这道题的时候,其实是想到了向左看和向右看两个方向的。具体来说,就是对于一栋楼i,去计算朝左边看能看到多少栋楼(或者说有多少栋楼被挡住)left[i],朝右边看能看到多少栋楼 right[i]。这个思路是没有问题的。问题出在,当时误以为可以根据 i 和 i - 1 高度关系,来根据left[i]得到left[i + 1],结果后来发现这是行不通的。后来去查了答案,知道了单调栈这种数据结构,因此这道题就迎刃而解了。
单调栈,简单来说(以降序为例),就是对于某个元素,尝试入栈,若栈顶元素比要入栈的元素小,则弹出,再去比较栈顶元素和要入栈的元素,重复这个过程直到栈顶元素比要入栈的元素大,或者栈空,这个元素再入栈。
那么,单调栈能否解决这个问题呢?降序单调栈内的元素都是降序排列的,栈顶最小栈底最大,因此从栈顶到栈底的方向看过去,所有楼都是可见的,没有哪栋楼会被挡住。 而弹出栈的楼,肯定是会被要入栈的楼挡住的,因为要入栈的楼比弹出的楼高。也就是说,弹出栈的楼,以后是肯定不会被看见的了,可以放心大胆地丢弃。
搞清楚这两点,就很容易得知,对于某栋要入栈的楼来说,当前栈的大小,就是这栋楼向左看可以看到的楼的数目。然后我们再把比这栋楼矮的所有楼一个一个弹出,再把这栋楼入栈(也就是维护单调栈的性质)(注意这里一定要先计算栈的大小再弹出栈,因为比当前的楼要矮的楼,还是可以被当前的楼看见的,只是后面的楼看不见了),就可以考虑下一栋楼的情况了。实现的代码如下:
#include <iostream>