逛街(单调栈解法)

题目描述

小明在周末的时候和他的小伙伴来到大城市逛街，一条步行街上有很多高楼，共有 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='')

题目来源：https://kamacoder.com/problempage.php?pid=1033