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 |