[백준] 10828 스택 Python

1. 문제

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

https://www.acmicpc.net/problem/10828

2. 알고리즘 분류

* 구현 

* 자료 구조

* 스택

 

3. 접근 방식

키워드 공부하면서 Stack 개념 에 대해 정리해 둔 것이 도움이 되었다

그리고 pop() 을 만들 때 그대로 .pop() 을 써도 되지만, 이왕 만드는 거 다른 방법으로 만들어 보고 싶어서

stack[-1] 을 다른 변수 lastpang 에 저장하고, stack[-1] 을 del 로 지운 후,

그 변수 lastpang 을 출력하는 방식으로 구현해 보았다

def pop():
    # print(stack.pop() if stack else -1)
    if stack:
        lastpang = stack[-1]
        del stack[-1]
        print(lastpang)
    else:
        print(-1)

 

4. 전체 코드

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

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

stack = []

def push(x):
    stack.append(x)

def pop():
    # print(stack.pop() if stack else -1)
    if stack:
        lastpang = stack[-1]
        del stack[-1]
        print(lastpang)
    else:
        print(-1)

def size():
    print(len(stack))

def empty():
    print(0 if stack else 1)

def top():
    print(stack[-1] if stack else -1)

if __name__ == '__main__':
    n = int(input())

    for _ in range(n):
        c = input().split()

        if c[0] == 'push':
            push(int(c[1]))
        elif c[0] == 'pop':
            pop()
        elif c[0] == 'size':
            size()
        elif c[0] == 'empty':
            empty()
        elif c[0] == 'top':
            top()