[python] 백준 - 10773. 괄호

🤔문제 해결

• S4 | 스택

너무 기본적인 스택문제라서 포스팅을 안하려고 했지만

input 과 sys.stdin.readline 에 대해서 말하고 싶어서 포스팅했다.

위의 사진을 보면 시간에서 큰 차이가 있을 것이다.

첫번째는 sys.stdin.readline을 사용하고, 두 번째는 input을 사용했다.

엄청난 시간차이가 나타난다.

이렇게 한줄 한줄 읽는 식으로 인풋값을 받을 때는 첫번째 방법을 사용하는 게 훨씬 빠르다.
속도 비교 하기

[python] input, sys.stdin.readline

문제 해결 방법은

1. 빈 배열을 하나 만든다.
2. 받은 값이 0이 아닌경우 배열에 값을 추가한다. append
3. 받은 값이 0인 경우 배열에서 값을 하나 뺀다. pop

문제에서 항상 pop을 보장한다고 했으므로 빈 배열에서 pop을 할 수도 있는 예외처리를 안해줘도 된다.

💻소스 코드

import sys

input = sys.stdin.readline
K = int(input())
stack = []
for _ in range(K):
    n = int(input())
    if n:
        stack.append(n)
    else:
        stack.pop()

print(sum(stack))
 

 

📕문제 확인

출처: BACKJOON ONLINE JUDGE

링크: https://www.acmicpc.net/problem/10773

10773번: 제로