[알고리즘] 백준 10828 (파이썬 풀이)

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

10828번: 스택

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

• push X: 정수 X를 스택에 넣는 연산이다.
• pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 스택에 들어있는 정수의 개수를 출력한다.
• empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
• top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

 

해결코드

import sys

n = int(sys.stdin.readline())
li = []
for i in range(n):
    m = sys.stdin.readline().split()
    if(m[0]=='push'):
        li.append(m[1])
    elif(m[0]=='pop'):
        if(len(li)==0):
            print(-1)
        else:
            print(li.pop(-1))
    elif(m[0]=='size'):
        print(len(li))
    elif(m[0]=='empty'):
        if(len(li)==0):
            print(1)
        else:
            print(0)
    elif(m[0]=='top'):
        if(len(li)==0):
            print(-1)
        else:
            print(li[-1])

 

문제풀이

문제 조건대로 식을 작성하면 그렇게 어렵지 않은 문제이다.

입력시간을 줄이기 위하여 sys라이브러리의 readline을 사용하여 n을 입력받는다.

그리고 스택을 표현하는 리스트를(li) 하나 만들어준다.

n만큼 for 반복문을 돌린다.

push 1 같은 문자열을 받기위하여 split()을 써서 m을 입력받는다.

만약 m[0]이 'push'라면 li에 m[1] 값을 추가해준다.

m이 'pop'일때 li의 크기가 0이라면 -1을 출력하고 아니라면 li의 가장 최근 값을 제거하고

그 값을 출력해준다.

m이 'size'라면 li의 길이 값을 출력해주고

m이 'empty'일때 li가 비어있다면 1, 비어있지 않다면 0을 출력해준다.

마지막으로 m이 'top'일때 li의 길이가 0이라면 -1을 출력해주고

아니라면 li에서 가장 위에 있는 값을 출력해준다.