문제
https://www.acmicpc.net/problem/18258
18258번: 큐 2
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 2,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지
www.acmicpc.net
풀이
from collections import deque
import sys
input=sys.stdin.readline
queue=deque([])
n = int(input())
for i in range(n):
arr = input().split()
if arr[0]=="push": queue.append(arr[1])
elif arr[0]=="pop": print(queue.popleft()) if queue else print(-1)
elif arr[0]=="size": print(len(queue))
elif arr[0]=="empty": print(0) if queue else print(1)
elif arr[0]=="front": print(queue[0]) if queue else print(-1)
else: print(queue[-1]) if queue else print(-1)
큐를 사용해서 풀면된다. 파이썬에서는 queue의 시간을 줄이기 위해서 deque를 사용할 수 있다.
느낀점
코테에 이정도 수준 문제만있다면 얼마나 좋을까.. 더 열심히 하자,,
'코딩테스트 > Python' 카테고리의 다른 글
| [Python] 백준 #11866 - 요세푸스 문제 0 (1) | 2021.12.07 |
|---|---|
| [Python] 백준 #2164 - 카드2 (1) | 2021.12.07 |
| [Python] 백준 #1874 - 스택 수열 (1) | 2021.12.05 |
| [Python] 백준 #4949 - 균형잡힌 세상 (0) | 2021.12.03 |
| [Python] 백준 #9012 - 괄호 (0) | 2021.12.02 |