[python/파이썬] 백준 10828 - 스택(실4)

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

10828번: 스택

---

자료구조 - 스택(stack)

 

파이썬에서는 따로 stack이라는 라이브러리? 모듈은 없고, 리스트를 사용하면 된다.

stack은 '후'입'선'출 형태의 자료구조이며, 차곡차곡 쌓인 것을 위에서부터 덜어낸다고 생각하면 된다.

---

문제를 풀 때 함수(def)를 만들거나, class를 생성하여 작성하고 싶었는데

가변적인 파라미터를 받을 때 적용 방법이나, 

class 생성 등에서 부족한 부분이 많아 포기.. ㅜㅜ

---

import sys
input = sys.stdin.readline

n = int(input()) 
stack = []
for _ in range(n):
    input_ord = input().split() # 자동으로 리스트로 받아낸다.
    # print(input_ord)
    order = input_ord[0]
    # print(order)
    
    if order == 'push':
        stack.append(input_ord[1])
        
    elif order == 'pop':
        if len(stack) == 0:
            print(-1) # pop으로 뺀 요소에 대해서 프린트 해줘야하므로 맨 마지막 인덱스를 입력해줘야 한다.
        else:
            print(stack.pop(-1))
            
    elif order == 'size':
        print(len(stack))
        
    elif order == 'empty':
        if len(stack) == 0:
            print(1)
        else:
            print(0)
            
    elif order == 'top':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack[-1])

 

---

[ 함수 작성 코드 (적용 실패..) ]

해당 코드를 먼저 작성하고 함수를 활용하여 입력 - 출력하고 싶었는데 난관에 봉착하여 그냥 if문으로 돌려서 풀었다..

# 함수로 만들어서 해보기
def stack(x, y):
    s = []
    if x == 'push':
        return s.append(y)
    elif x == 'pop':
        if len(s) != 0:
            return s.pop(-1)
        else:
            return -1
    elif x == 'size':
        return len(s)
   
    elif x == 'empty':
        if len(s) != 0:
            return 0
        else:
            return 1
       
    elif x == 'top':
        if len(s) != 0:
            return s[-1]
        else:
            return -1