10828번 스택

문제링크 : www.acmicpc.net/problem/10828

10828번: 스택

---

---

버전 1. if문을 이용한 분기 ( 메모리 : 28776KB    시간 : 72ms)

import sys
N = int(sys.stdin.readline())
numbers = []
for _ in range(N):
    commend = sys.stdin.readline().split()
    if(commend[0] == "push"):
        num = int(commend[1])
        numbers.insert(0,num)
    elif(commend[0]=="pop"):
        if(len(numbers)==0):
            print(-1)
        else:
            print(numbers[0])
            del numbers[0]
    elif (commend[0] == "size"):
        print(len(numbers))
    elif (commend[0] == "empty"):
        if(len(numbers) ==0):
            print(1)
        else :
            print(0)
    elif(commend[0]=="top"):
        if (len(numbers) == 0):
            print(-1)
        else:
            print(numbers[0])

 

 

버전 2. 함수를 이용 ( 메모리 : 28776KB    시간 : 72ms)

numbers = []
def size():
    print(len(numbers))
def push(a):
    numbers.insert(0,a)
def pop():
    if(len(numbers)==0):
        print(-1)
    else:
        print(numbers[0])
        del numbers[0]
def empty():
    if(len(numbers)==0):
        print(1)
    else:
        print(0)
def top():
    if (len(numbers) == 0):
        print(-1)
    else:
        print(numbers[0])
        
import sys
N = int(sys.stdin.readline())
for _ in range(N):
    commend = sys.stdin.readline().split()
    if(commend[0] == "push"):
        push(commend[1])
    elif(commend[0] == "pop"):
        pop()
    elif (commend[0] == "size"):
        size()
    elif (commend[0] == "empty"):
        empty()
    elif (commend[0] == "top"):
        top()