BOJ_17608 : 막대기

👉 문제링크

17608번: 막대기

---

---

🔸 문제 분석 🔸

• 보는 방향에서 봤을때 보이는 블럭의 수를 출력한다.
• 스택을 사용해서 가려지는 블록을 제거하면 되는 문제이다.

🔸 코드 🔸

import sys

input = sys.stdin.readline

N = int(input())

arr = [0]

for i in range(N):
    h = int(input())
    while True:
        if len(arr) == 0:
            arr.append(h)
            break
        elif arr[-1] <= h:
            arr.pop()
        else:
            arr.append(h)
            break
print(len(arr))

🔸 코드 해석 🔸

• 리스트를 스택으로 사용했다.
• 문제 시간제한으로 빠른 입력을 사용했다.
• 스택의 가장 위 값보다 큰 값이 들어오면 pop한다.

---

🔸 end 🔸

• 시간초과가 떠서 빠른입출력을 추가했는데, 문제 조건을 보고 미리 대비할 필요가 있어보인다.
  (초과되는지 궁금해서 눌러보게 된다..)
• while문 안의 if과 else문을 합치면 더 좋았을 것 같다.