백준 | [파이썬 Python] 옥상 정원 꾸미기 6198

---

문제

https://www.acmicpc.net/problem/6198

 

 

코드

n = int(input())
stack = []
answer = 0

for _ in range(n):
    h = int(input())
    while stack and stack[-1] <= h:
        stack.pop()
    answer += len(stack)
    stack.append(h)
print(answer)

 

 

설명

볼 수 있는 위치를 stack에 담아서 처리하는 코드 이다.

코드는 간단하지만 생각의 틀을 깨야 풀 수 있는 문제였다