题解 | #枪打出头鸟#

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param n int整型 n个人
# @param a int整型一维数组 ai代表第i个人的高度
# @return long长整型
#
class Solution:
    def solve(self , n , a ):
        # write code here
        ans = 0
        stack = []
        for i in range(n):
            while stack and a[i] >= a[stack[-1]]:
                stack.pop()
            if stack:
                ans += stack[-1] + 1
            stack.append(i)
        return ans