티스토리 뷰

알고리즘

백준 10866번 덱 파이썬

killog 2020. 9. 30. 21:46
반응형

 파이썬 덱을 구현해 보는 기회였다. 시간 초과때문에 빡쳤지만, sys.stdin.readline() 에서 split(" ") 가 아닌 split() 만 작동함을 깨우쳤다.

 

문제

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

명령은 총 여덟 가지이다.

  • push_front X: 정수 X를 덱의 앞에 넣는다.
  • push_back X: 정수 X를 덱의 뒤에 넣는다.
  • pop_front: 덱의 가장 앞에 있는 수를 빼고, 그 수를 출력한다. 만약, 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
  • pop_back: 덱의 가장 뒤에 있는 수를 빼고, 그 수를 출력한다. 만약, 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
  • size: 덱에 들어있는 정수의 개수를 출력한다.
  • empty: 덱이 비어있으면 1을, 아니면 0을 출력한다.
  • front: 덱의 가장 앞에 있는 정수를 출력한다. 만약 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
  • back: 덱의 가장 뒤에 있는 정수를 출력한다. 만약 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

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

출력

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

import collections
import sys
N = int(sys.stdin.readline())
dec= collections.deque([])
for _ in range(N):
    commands = list(sys.stdin.readline().split()) # sys.stdin.readline() 에서 split 은 " "가 작동하지 않는다.
    if commands[0]=="push_back":
        dec.append(int(commands[1]))
        continue
    elif  commands[0]=="push_front":
        dec.appendleft(int(commands[1]))
        continue
    elif commands[0]=="pop_back":
        if len(dec)>0:
            print(dec.pop())
        else:
            print(-1)
        continue
    elif  commands[0]=="pop_front":
        if len(dec)>0:
            print(dec.popleft())
        else:
            print(-1)
        continue
    elif  commands[0]=="front":
        if len(dec)>0:
            print(dec[0])
            
        else:
            
            print(-1)
        continue
    elif  commands[0]=="back":
        if len(dec)>0:
            print(dec[-1])
        else:
            print(-1)
        continue
    
    elif  commands[0]=="size":
            print(len(dec))
            continue
    elif  commands[0]=="empty":
            print(int(len(dec)==0))
            continue
반응형
댓글
반응형
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
«   2024/12   »
1 2 3 4 5 6 7
8 9 10 11 12 13 14
15 16 17 18 19 20 21
22 23 24 25 26 27 28
29 30 31
글 보관함