[ 파이썬(python) ] 백준 17608 - 막대기

📍 백준 17608 - 막대기

문제: 백준 17608 - 막대기

---

💡 나의 풀이

일렬로 세워진 막대기를 오른쪽에서 봐야 하기 때문에 막대기의 입력을 모두 받아 리스트로 만들고 문제를 풀었다. 처음에는 오답 판정을 받았는데 최대 높이의 막대기를 갱신해주지 않아서 그랬다.

 

첫번째 방법은 sticks인덱스를 거꾸로 확인하여 최대의 막대기를 갱신한 방법이고 두 번째 방법은 sticks를 pop()으로 하나씩 뽑아 최댓값을 갱신하는 방법을 사용했다. pop()을 사용한 방법의 실행시간이 약 30m/s정도 빨랐다.

 

 

(사진에서 첫번째 방법은 하단, 두 번째 방법은 상단)

# 첫번째 방법
import sys
input = sys.stdin.readline

n = int(input())
sticks = [int(input()) for _ in range(n)]
max_height = sticks[-1]
cnt = 1

for i in range(n):
    if max_height < sticks[n-i-1]:
        cnt += 1
        max_height = sticks[n-i-1]
print(cnt)

# 두번째 방법
import sys
input = sys.stdin.readline

n = int(input())
sticks = [int(input()) for _ in range(n)]
max_height = sticks[-1]
cnt = 1

for i in range(n):
    temp = sticks.pop()
    if max_height < temp:
        cnt += 1
        max_height = temp
print(cnt)