[Python] 백준 17608 : 막대기

스택 문제 중 하나인 막대기를 풀어보았다.

 

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

17608번: 막대기

 

스택에 다 넣어주고 빼면서 높은 건물이 나올때 기준 높이를 바꿔주면 될 것 같다.

오른쪽부터 시작하니 맨 오른쪽을 기준 높이로 잡고 왼쪽으로 하나씩 확인해 나간다.

 

import sys

input = sys.stdin.readline

N = int(input())
stack = []
cnt = 1
for _ in range(N):
    num = int(input())
    stack.append(num)
right_stick = stack[N - 1]
idx = N - 2
while True:
    if right_stick < stack[idx]:
        cnt += 1
        right_stick = stack[idx]
    if idx == -1:
        break
    idx -= 1
print(cnt)

 

처음 제출 시 시간초과가 났었는데 input을 readline으로 바꿔주니 해당 이슈는 없어졌다.

 

오늘도 감사합니다.