[백준]스택

2020. 9. 4. 21:161일 1 알고리즘

이번에는 정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 알고리즘을 작성합니다.

n = int(input())
a=[]

for i in range(n):
    cmd = input().split()
    if cmd[0] == 'push':
        a.append(cmd[1])
    elif cmd[0] == 'pop':
        try:print(a.pop())
        except: print(-1)
    elif cmd[0] == 'size':
        print(len(a))
    elif cmd[0] == 'empty':
        if len(a)==0:
            print(1)
        else:
            print(0)
    elif cmd[0] == 'top':
        try:
            print(a[-1])
        except:
            print(-1)

 처음에는입력할 변수와 배열을 만들어주고 n번 만큼 반복시켜줍니다 다음 각각의 이름을 지정해주고 if문을 사용하여 명령이 무엇인지 확인하여주고 출력해줍니다

'1일 1 알고리즘' 카테고리의 다른 글

[백준]괄호  (0) 2020.09.08
[백준]팩토리얼 0의 개수  (0) 2020.09.02
[백준]이항 계수 2  (0) 2020.09.01
[백준]이항 계수 1  (0) 2020.08.31
[백준]패션왕 신혜빈  (0) 2020.08.28