题目描述
小明在周末的时候和他的小伙伴来到大城市逛街,一条步行街上有很多高楼,共有 n 座高楼排成一行。
小明从第一栋一直走到了最后一栋,小明从来都没有见到这么多的楼,所以他想知道他在每栋楼的位置处能看到多少栋楼呢?(当前面的楼的高度大于等于后面的楼时,后面的楼将被挡住)
输入描述
输入一个数组,表示每栋楼的高度。
输出描述
输出一个数组,表示在每栋楼对应的位置上能看到多少栋楼。
输入示例
[5,3,8,3,2,5]
输出示例
[3,3,5,4,4,4]
提示信息
当小明处于位置 3 时,他可以向前看到位置 2, 1 处的楼,向后看到位置 4, 6 处的楼,加上第 3 栋楼,共可看到 5 栋楼。
当小明处于位置 4 时,他可以向前看到位置 3 处的楼,向后看到位置 5, 6 处的楼,加上第 4 栋楼,共可看到 4 栋楼。
1 < n < 1000.
答案解析:利用单调栈,先从前往后,再从后往前
def walk(a1):
n=len(a1)
if n==1:
return [1]
pre=[1]*n
post=[1]*n
# 利用单调栈
s=[]
for i in range(n):
while s and a1[i]>=a1[s[-1]]:
s.pop()
pre[i]=len(s)+1
s.append(i)
s=[]
for i in range(n-1,-1,-1):
while s and a1[i]>=a1[s[-1]]:
s.pop()
post[i]=len(s)+1
s.append(i)
ans=[1]*n
ans[0]=1+post[1]
ans[-1]=1+pre[-2]
for i in range(1,n-1):
ans[i]=pre[i-1]+post[i+1]+1
return ans
data=input()[1:-1]
data=list(map(int,data.split(',')))
ans=walk(data)
print('[' + ','.join(map(str, ans)) + ']', end='')