[백준 10828] 스택(Feat. Python)

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

• push X: 정수 X를 스택에 넣는 연산이다.
• pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 스택에 들어있는 정수의 개수를 출력한다.
• empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
• top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

 

풀이

• 위 문제는 결국에는 기본적인 stack의 구조를 아는지 모르는지에 대한 문제라고 이해했고 이를 통해 명령어를 내렸을때 해당 함수를 실행 시키면 되겠다 라는 생각을 했다.
• 본인은 answer 배열을 전역변수로 선언해서 출력해야하는 답변을 만들었다.

 

answer = []
def isEmpty(stack):
    answer.append(1 if (len(stack) == 0) else 0)
def push(stack, X):
    stack.append(X)
def size(stack):
    answer.append(len(stack))
def top(stack):
    if(len(stack) == 0):
        answer.append(-1)
    else:
        answer.append(stack[-1])
def pop(stack):
    if (len(stack)== 0):
        answer.append(-1)
    else:
        answer.append(stack.pop())

N = int(input())
stack = []

for _ in range(N):
    str1 = input()
    if(str1.find('push') != -1):
        arr = str1.split(' ')
        push(stack, int(arr[1]))
    elif(str1.find('top') != -1):
        top(stack)
    elif(str1.find('size') != -1):
        size(stack)
    elif(str1.find('empty') != -1):
        isEmpty(stack)
    elif(str1.find('pop') != -1):
        pop(stack)

for i in range(len(answer)):
    print(answer[i])

• find 함수를 사용하여 입력된 값에서 push, top, size, pop, empty 글자를 찾을 수 있으면 (저 함수의 결과값이 -1이 아니면) 해당 if문이 실행되도록 하였다.