백준 - 스택 10828

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

10828번: 스택

문제 설명

- push와 pop과 size와 empty와 top을 구현하면 되는 문제

 

코드 구상

1. push는 문자열 인덱스 5번 이후의 값을 넣는다

2. pop은 빈 리스트일 경우에 -1를 출력하고, 아닐경우엔 첫번째 요소를 불러오고 삭제한다

3. top은 빈 리스트일 경우에 -1를 출력하고, 아닐경우엔 첫번째 요소를 불러온다

4. size는 리스트의 크기를 출력한다

5. empty는 빈 리스트일 경우 1을 출력하고 아니면 0을 출력한다

 

코드

n = int(input())
arr = [input() for _ in range(n)]
stack = []
for i in arr:
    if i == 'pop': 
        if len(stack)==0:
            print(-1)
        else:
            print(stack[0])
            del stack[0]
    elif i == 'top':
        if len(stack) ==0:
            print(-1)
        else:
            print(stack[0])
    elif i[0:4] == 'push':
        stack.insert(0,i[5:])
    elif i == 'size':
        print(len(stack))
    elif i =='empty':
        if len(stack) ==0:
            print(1)
        else:
            print(0)