维护一个单调栈,开始遍历所有人的高度,当当前遍历到的人的高度,高于单调栈中栈顶的人的高度,意味着,当前的人是栈顶的人可以看到的最远的一个人,此时,栈顶的人可以看到的人数+1,随后就可以丢了,继续把新的栈顶的人和当前的人最比较,将新栈顶的人看到的人再+1,不高过当前的人就弹出,高过就保留,随后把新的当前人的位置如栈。单调栈中,每当有新的人,都需要把栈自顶向下,把比当前人高度低的人都弹出。
class Solution:
def canSeePersonsCount(self, heights: List[int]) -> List[int]:
res = [0 for _ in range(len(heights))]
monostack = []
for i in range(len(heights)):
while monostack and heights[i] > heights[monostack[-1]]:
res[monostack[-1]] += 1
monostack.pop()
if monostack:
res[monostack[-1]] += 1
monostack.append(i)
return res