Introfor

[백준] 10866번 덱 본문

카테고리 없음

[백준] 10866번 덱

YongArtist 2020. 9. 12. 21:08
from sys import stdin
from collections import deque

if __name__ == '__main__':
    n = int(stdin.readline())
    Deque = deque()

    for _ in range(n):
        cmd = stdin.readline().split()

        if cmd[0] == 'push_front':
            Deque.appendleft(cmd[1])

        elif cmd[0] == 'push_back':
            Deque.append(cmd[1])

        elif cmd[0] == 'pop_front':
            if len(Deque) == 0:
                print(-1)
            else:
                print(Deque.popleft())

        elif cmd[0] == 'pop_back':
            if len(Deque) == 0:
                print(-1)
            else:
                print(Deque.pop())

        elif cmd[0] == 'size':
            print(len(Deque))

        elif cmd[0] == 'empty':
            if len(Deque) == 0:
                print(1)
            else:
                print(0)

        elif cmd[0] == 'front':
            if len(Deque) == 0:
                print(-1)
            else:
                print(Deque[0])

        elif cmd[0] == 'back':
            if len(Deque) == 0:
                print(-1)
            else:
                print(Deque[-1])

원래 이 코드를 작성할 때 list로 작성했는데, collections 모듈에 deque를 확인하고 바꿔서 작성했다.

큐를 구현하면서도 list.insert()로 값을 삽입할 때 O(n)이라 마음이 불편했는데 오늘 딱 개비스콘 먹은 기분이다.

collections.deque 시간 복잡도 관련 사이트는 아래 링크에 있다.

daimhada.tistory.com/56

Comments