728x90
반응형
문제
정수를 저장하는 덱(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보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
예제 입력1
15
push_back 1
push_front 2
front
back
size
empty
pop_front
pop_back
pop_front
size
empty
pop_back
push_front 3
empty
front
예제 출력1
2
1
2
0
2
1
-1
0
1
-1
0
3
# Code
import sys
from collections import deque
n = int(sys.stdin.readline()) #input으로 하면 시간초과
deque = deque()
for _ in range(n):
s = sys.stdin.readline().split()
if s[0] == 'push_front':
deque.appendleft(s[1])
elif s[0] == 'push_back':
deque.append(s[1])
elif s[0] == 'pop_front':
if len(deque)>0: print(deque.popleft())
else: print(-1)
elif s[0] == 'pop_back':
if len(deque)>0: print(deque.pop())
else: print(-1)
elif s[0] == 'size':
print(len(deque))
elif s[0] == 'empty':
if len(deque)==0: print(1)
else: print(0)
elif s[0] == 'front':
if len(deque)>0: print(deque[0])
else: print(-1)
elif s[0] == 'back':
if len(deque)>0: print(deque[-1])
else: print(-1)
# Comment
앞에서 풀었던 stack, queue 문제의 deque 버전이다. 따라서 부가적인 설명을 추가할 필요는 없을 듯 싶다.
여기서 한 번 실수했던건.. deque 설명 대충 읽어보고 값 추가하려면 extend 쓰면 되는건가? 싶어서 그냥 막 쓰고 제출했는데 실패라고 떠서 살짝 당황했다. 근데 알고보니 extend는 결이 다른 놈이었군..
- append와 extend의 차이
deque = deque()
deque.append('b')
# deque = ['b']
deque.extend('lt')
# deque = ['b', 'l', 't']
extend는 문자를 찢어서(?) 뒤에 하나씩 붙여주는 놈이었던 것이다........ 이렇게 또 하나 알아간다....ㅎㅎ...
728x90
반응형
'백준 > 자료구조' 카테고리의 다른 글
[백준] #1935 후위 표기식2 (python) (0) | 2022.02.08 |
---|---|
[백준] #1874 스택수열 (python) (0) | 2022.02.08 |
[백준] #2164 카드 2 (python) (2) | 2022.02.05 |
[백준] #18258 큐 2 (python) (0) | 2022.02.04 |
[백준] #9012 괄호 (python) (0) | 2022.02.04 |
댓글