문제
정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 다섯 가지이다.
push X: 정수 X를 스택에 넣는 연산이다.
pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
size: 스택에 들어있는 정수의 개수를 출력한다.
empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다.
해결 방안
파이썬 내장함수를 사용해 쉽게 해결했다!!
명령어와 원소 입력은 한 줄로 받은 후에 split을 활용해 명령어는 0번째 공간에 담고 원소는 1번째 공간에 담았다.
push : stack.append()
pop : stack.pop()
size, empty : len(stack)
top : stack[-1]
코드
import sys
n = int(input())
stack = []
for i in range(n):
command = sys.stdin.readline().split()
if command[0] == "push":
stack.append(int(command[1]))
elif command[0] == "pop":
if len(stack) == 0:
print("-1")
else:
top = stack.pop()
print(top)
elif command[0] == "size":
print(len(stack))
elif command[0] == "empty":
if len(stack) == 0:
print("1")
else:
print("0")
elif command[0] == "top":
if len(stack) == 0:
print("-1")
else:
print(stack[-1])스택이란?
배열로 스택을 구현할 경우 위 그림처럼
push 연산은 가장 마지막에 원소를 넣고
pop 연산은 가장 마지막 원소, 즉 최근에 넣은 원소를 뺀다.
top 연산 역시 가장 최근에 넣은 원소를 출력한다.
C언어로 스택을 구현할 때는 귀찮았는데 파이썬은 내장 함수들이 많아서 확실히 편하게 풀 수 있었다ㅋㅋㅋㅋ
그런데 처음 문제를 풀고 제출했더니 시간 초과 오류가 발생했다.
처음에 명령어 입력을 command = input().split() 으로 작성했다.
알아보니 input()이 시간을 많이 쓴다고 했다.
그래서 입력 코드를 command = sys.stdin.readline().split() 과 같이 고치니 시간 초과 오류가 발생하지 않았다.
대체 무슨 차이인지 모르겠지만 이게 가장 빠르다고 하니 앞으로는 이걸 쓰도록 해야겠다;;👍😊👍😊
