본문 바로가기

알고리즘

백준-10866

https://www.acmicpc.net/problem/10866

 

10866번: 덱

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

www.acmicpc.net

처음에는 이 문제를 큐를 사용하지 않고 아래처럼 풀었습니다

cnt = int(input())
result = []
for i in range(cnt):
    qus = list(input().split())
    if qus[0] == 'push_front':
        result.insert(0, qus[1])
    elif qus[0] == 'push_back':
        result.append(qus[1])
    elif qus[0] == 'pop_front':
        if len(result) == 0:
            print(-1)
        else:
            first = result.pop(0)
            print(first)
    elif qus[0] == 'pop_back':
        if len(result) == 0:
            print(-1)
        else:
            last = result.pop()
            print(last)
    elif qus[0] == 'size':
        print(len(result))
    elif qus[0] == 'empty':
        if len(result) == 0:
            print(1)
        else:
            print(0)
    elif qus[0] == 'front':
        if len(result) == 0:
            print(-1)
        else:
            print(result[0])
    else:
        if len(result) == 0:
            print(-1)
        else:
            print(result[-1])

정상적으로 동작을 하지만 시간초과가 발생하여서 deque라이브러리를 사용하여 문제를 해결하였습니다

from collections import deque
import sys

d = deque()
n = int(input())

for i in range(n):
    command = sys.stdin.readline().split()

    if command[0] == "push_front":
        d.appendleft(command[1])
    elif command[0] == "push_back":
        d.append(command[1])
    elif command[0] == "pop_front":
        if d:
            print(d[0])    
            d.popleft()
        else:
            print("-1")
    elif command[0] == "pop_back":
        if d:
            print(d[len(d) - 1])    
            d.pop()
        else:
            print("-1")
    elif command[0] == "size":
        print(len(d))
    elif command[0] == "empty":
        if d:
            print("0")
        else:
            print("1")
    elif command[0] == "front":
        if d:
            print(d[0])
        else:
            print("-1")
    elif command[0] == "back":
        if d:
            print(d[len(d) - 1])
        else:
            print("-1")

push_front:

d.appendleft(command[1])

input형태 : push_front 3

command[1]인 3을 큐 맨앞에 삽입해준다.

push_back:

d.append(command[1])

input형태 : push_front 3

command[1]인 3을 큐 맨뒤에 삽입해준다.

pop_front:
큐가 존재하면 맨앞을 pop(popleft) 해주고 비어있으면 -1 print

pop_back:

큐가 존재하면 맨뒤를 print해주고 pop(맨뒤에 제거) 비어있으면 -1 print

size:

큐의 len 길이를 출력

empty:
큐의 길이가 0이면 0, 0이 아니면 1 print

front:

큐가 존재하면 맨앞 print 존재하지않으면 -1 print

back:

큐가 존재하면 맨뒤 print 존재하지 않으면 -1 print

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

[프로그래머스 LV 2] 예상 대진표(파이썬)  (2) 2023.12.07
백준-1158 (파이썬)  (0) 2023.10.02
이진 탐색 - 구현  (0) 2023.09.25
퀵 정렬  (0) 2023.09.22
BFS) 미로 탈출  (0) 2023.09.22