BOJ_10828 : 스택

👉 문제링크

10828번: 스택

---

---

🔸 문제 분석 🔸

• 스택을 구현한다.
• 시간 제한이 0.5로 짧다.

🔸 코드 🔸

import sys
input = sys.stdin.readline
print = sys.stdout.write

N = int(input().strip())
arr = []

for i in range(N):
    order = input().strip()
    
    if order[:4] == "push":
        arr.append(int(order[5:]))
    elif order == "pop":
        if(len(arr) == 0): print(str(-1)+'\n')
        else:
            print(str(arr[-1])+'\n')
            arr.pop()
    elif order == "size":
        print(str(len(arr))+'\n')
    elif order == "empty":
        if(len(arr) == 0): print(str(1)+'\n')
        else: print(str(0)+'\n')
    elif order == "top":
        if(len(arr) == 0): print(str(-1)+'\n')
        else: print(str(arr[-1])+'\n')

🔸 코드 해석 🔸

• 빠른 입출력을 사용했다.
• sys.stdin.readline 의 경우 개행문자(\n)까지 읽으므로 strip() 함수로 제거했다.
• sys.stdout.write 의 경우 int값을 str로 바꾸어 출력하고 개행 문자도 추가했다.
• 어떤 명령인지 if, elif로 구분했다.
• 입력값을 슬라이스를해 push와 숫자를 구분했다.

---

🔸 end 🔸

• push만 숫자가 딸려있어서 구분 방법을 조금 고민했다.
• 시간초과가 떠서 빠른 입출력을 사용했고, 그덕에 코드가 좀 지저분해진 것 같다.