[백준][Python] 10828 스택

구현, 자료구조, 스택 / 실버 4

📚

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

10828번: 스택

 

 

내 풀이

import sys
N = int(sys.stdin.readline())
stack = []

for _ in range(N):
    command = list(sys.stdin.readline().split())

    if command[0] == 'push':
        stack.append(command[1])
        
    elif command[0] == 'pop':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack.pop(-1))

    elif command[0] == 'size':
        print(len(stack))

    elif command[0] == 'empty':
        if len(stack) == 0:
            print(1)
        else:
            print(0)

    elif command[0] == 'top':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack[-1])

스택 기본 개념 정리하기 좋은 문제!

 

 

코드 설명

 

1. push

스택에 정수 X 추가

if command[0] == 'push':
    stack.append(command[1])

 

2. pop

- 가장 위에 있는 정수(가장 마지막에 추가된 정수)를 뺀 후 출력

- 스택이 비어있으면 -1 출력

elif command[0] == 'pop':
    if len(stack) == 0:
        print(-1)
    else:
        print(stack.pop(-1))

 

3. size

스택에 들어있는 정수의 개수 출력

elif command[0] == 'size':
    print(len(stack))

 

4. empty

- 스택이 비어있으면 1 출력

- 비어있지 않으면 0 출력

elif command[0] == 'empty':
    if len(stack) == 0:
        print(1)
    else:
        print(0)

 

5. top

- 가장 위에 있는 정수(가장 마지막에 추가된 정수)를 출력

- 스택이 비어있으면 -1 출력

elif command[0] == 'top':
    if len(stack) == 0:
        print(-1)
    else:
        print(stack[-1])

 

 

🧸 참고 사항 🧸

 

1. 스택의 마지막 값을 가져올 때, 처음엔 stack[len(stack)-1]로 했었는데, stack[-1]로 쓰면 간단해진다.

2. 시간 초과가 발생했다. 사용자 입력을 받는 방식을 input() -> sys.stdin.readline()으로 변환하니 해결되었다.