10828 : 스택

서희찬·2021년 11월 16일
0

백준

목록 보기
73/105

문제

코드

import sys 
input = sys.stdin.readline 

stack=[]

n = int(input())
for i in range(n):
    choice = input().split()

    if choice[0]=='push':
        stack.append(choice[1])
    elif choice[0]=='pop':
        if len(stack)==0:
            print(-1)
        else :
            print(stack.pop())
    elif choice[0]=='size':
        print(len(stack))
    elif choice[0]=='empty':
        if len(stack)==0: #ㅂㅣ어ㅆㅑ? 
            print(1)
        else :
            print(0)
    elif choice[0]=='top':
        if len(stack)==0:
            print(-1)
        else :
            print(stack[-1]) #마지막요소 출력 

해설

파이썬에서는 스택형 자료구조는 리스트로 만들면된다!
그냥 리스트로 팝해주고 푸쉬해주고 뭐..그런식....
아주 잘되있다.
명령과 숫자를 받는 방식을 choice라는 문자열에 0,1번째에 저장하는 방식을 활용하면 이 문제를 해결할 수 있다.

profile
부족한 실력을 엉덩이 힘으로 채워나가는 개발자 서희찬입니다 :)

0개의 댓글