S4 10828. 스택

문제 보기 :  10828번: 스택

---

문제

• 정답률 : 38%

---

작성 코드

import sys
input = sys.stdin.readline

n = int(input().strip())
stack = []


for _ in range(n):
    word = input().rstrip()

    if word == 'pop':
        if not stack:
            print(-1)
        else:
            print(stack.pop())
    
    elif word == 'size':
        print(len(stack))
    
    elif word == 'empty':
        if stack:
            print(0)
        else:
            print(1)
    
    elif word == 'top':
        if not stack:
            print(-1)
        else:
            print(stack[-1])
        
    else:
        word, num = word.split()
        num = int(num)
        stack.append(num)

---

풀이

쉬운 문젠데

import sys
input = sys.stdin.readline
input().rstrip()
input().strip()

이 부분 때문에 정답률이 낮을 거라 생각했다.

rstrip()은 문자열 입력을 받을 때 사용하고,
strip()은 숫자 입력을 받을 때 사용한다.