BOJ10828-스택 (python3)

Ok Haeeun·2023년 3월 13일
0

Python3로 algorithm문풀

목록 보기
28/53

파이썬의 list에서 제공하는 메서드 그대로 사용해서 풀었다.
이래도 되나..?어쨌든 맞았음.
야무진 lstrip() 누가 만들었을까. 칭찬드립니다..

파이썬에서는 switch문을 제공하지 않기 때문에 if-elif-else문으로 비벼봐야한다는게 중요한 포인트라고 할 수 있겠다.

import sys

n = int(sys.stdin.readline())
stack_list = []
for i in range(n):
    str = sys.stdin.readline()
    if "push" in str:
        stack_list.append(int(str.lstrip('push ')))
    elif "pop" in str:
        if len(stack_list) > 0:
            print(stack_list.pop())
        else: print(-1)
    elif "size" in str:
        print(len(stack_list))
    elif "empty" in str:
        if len(stack_list) > 0:
            print(0)
        else: print(1)
    elif "top" in str:
        if len(stack_list) > 0:
            print(stack_list[-1])
        else: print(-1)
profile
tistory에 이어서 기록합니다 👉 https://i-m-okay.tistory.com/

0개의 댓글

관련 채용 정보