[백준] 17608번 - 막대기 (python)

문제

아래 그림처럼 높이만 다르고 (같은 높이의 막대기가 있을 수 있음) 모양이 같은 막대기를 일렬로 세운 후, 왼쪽부터 차례로 번호를 붙인다. 각 막대기의 높이는 그림에서 보인 것처럼 순서대로 6, 9, 7, 6, 4, 6 이다. 일렬로 세워진 막대기를 오른쪽에서 보면 보이는 막대기가 있고 보이지 않는 막대기가 있다. 즉, 지금 보이는 막대기보다 뒤에 있고 높이가 높은 것이 보이게 된다. 예를 들어, 그림과 같은 경우엔 3개(6번, 3번, 2번)의 막대기가 보인다.

N개의 막대기에 대한 높이 정보가 주어질 때, 오른쪽에서 보아서 몇 개가 보이는지를 알아내는 프로그램을 작성하려고 한다.

---

입력

첫 번째 줄에는 막대기의 개수를 나타내는 정수 N (2 ≤ N ≤ 100,000)이 주어지고 이어지는 N줄 각각에는 막대기의 높이를 나타내는 정수 h(1 ≤ h ≤ 100,000)가 주어진다.

출력

오른쪽에서 N개의 막대기를 보았을 때, 보이는 막대기의 개수를 출력한다.

---

예제 입력 1 복사

6 6 9 7 6 4 6

예제 출력 1 복사

3

예제 입력 2 복사

5 5 4 3 2 1

예제 출력 2 복사

5

---

풀이과정

n = int(input())
l = []

for _ in range(n):
	l.append(int(input()))
count = 0
max = 0
for x in reversed(l):
	if max < x:
		max = x
		count += 1
print(count)

역순으로 리스트를 돌려서 일일이 대입하는 형식으로 풀 수 있지않을까..? 라는 생각으로 대입을 했었지만

아쉽게도 시간초과 시간을 더 줄일 수 있는 방법을 생각해 봐야할거같다.

 

code

import sys
input = sys.stdin.readline

n = int(input())
l = []

for _ in range(n):
	l.append(int(input()))
count = 0
max = 0
for x in reversed(l):
	if max < x:
		max = x
		count += 1
print(count)

어떻게 작성을 해야할까 알고리즘을 어떻게 개선해야할까 고민을 했었는데

그냥 python 언어 자체에서 입력을 받는방식이 너무 느려서 생기는 문제였다.

입력받는 형식을 조금 더 개선해주었더니 해결 (약간 허무..)

 

후기쓰

python이라는 언어 자체는 작성하는 방법이 너무 깔끔하고 군더더기가 없어서 너무 마음에 드는데.

실제 프로그램을 작성할때나 컴파일을 할때 동작하는 방법이 c랑 달라서 시간적으로 차이가 너무 많이 나게된다.

속도를 생각한다면 다른 언어를 선택해야하지않을까라는 생각이 들게된다.