[백준] 28278번. 스택 2

[백준] 28278번. 스택 2 바로가기

아이디어

• 문제에서 하라는 대로 구현하면 됨

  1 X: 정수 X를 스택에 넣는다. (1 ≤ X ≤ 100,000)
  2: 스택에 정수가 있다면 맨 위의 정수를 빼고 출력한다. 없다면 -1을 대신 출력한다.
  3: 스택에 들어있는 정수의 개수를 출력한다.
  4: 스택이 비어있으면 1, 아니면 0을 출력한다.
  5: 스택에 정수가 있다면 맨 위의 정수를 출력한다. 없다면 -1을 대신 출력한다.

• 1번 명령어에서 예제만 보고 stack에 넣는 숫자가 한 자리수인 경우만 생각해서 틀림

시간 복잡도

• O(N)

코드

import sys
input = sys.stdin.readline

n = int(input())
command_list = [input().strip() for _ in range(n)]
stack = []

# 명령어
for command in command_list:

    if command == '2':
        if stack:
            print(stack.pop())
        else:
            print(-1)

    elif command == '3':
        print(len(stack))

    elif command == '4':
        if stack:
            print(0)
        else: 
            print(1)

    elif command == '5':
        if stack:
            print(stack[-1])
        else:
            print(-1)

    else:
        stack.append(int(command[2:]))