[백준/Python] 28278 스택 2

[백준/Python] 28278 스택 2

풀이 코드 및 설명

import sys 
input = sys.stdin.readline

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

for _ in range(N):
    command = input().rstrip()

    if len(command) > 2: #1일때 
        stack.append(int(command[2:]))
    elif command == '2': #2일때 
        if len(stack) == 0:
            print(-1)
        else:
            print(stack.pop())
    elif command == '3':
        print(len(stack))
    elif command == '4':
        print(1 if len(stack) == 0 else 0)
    elif command == '5':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack[-1])

1. stack = [] 스택 초기화
2. for 문 내 command 명령어 처리
   • if len(command) > 2 이 경우는 1번 x와 함께 입력 될 경우, 스택에 삽입하면 된다.
   • elif command == 2 명령어가 2일 때 맨 위 원소 제거 및 출력 pop 명령어 사용
   • elif command == 3 스택의 크기 출력
   • elif command == 4 스택이 비어있는 지 확인 비어있다면 1 아니면 0 을 출력한다
   • elif command == 5 스택의 맨 위 원소를 출력하지만 제거하지 않는다. 스택이 비어있다면 -1 을 출력한다.