[백준] 1021번 회전하는 큐 파이썬(Python) 풀이

1021번: 회전하는 큐

[문제]

---

[코드]

from collections import deque
import sys

n, m = map(int, sys.stdin.readline().split())
p = list(map(int, sys.stdin.readline().split())) #뽑으려는 수위치
dq = deque([i for i in range(1, n+1)])

count = 0

for i in p:
    while True:
        if dq[0] == i:
            dq.popleft()
            break
        else:
            if dq.index(i) < len(dq)/2:
                while dq[0] != i:
                    dq.append(dq.popleft())
                    count += 1
            else:
                while dq[0] != i:
                    dq.appendleft(dq.pop())
                    count +=1

print(count)

---

[해설]

데크(deque)를 활용한 간단한 구현 문제이다.

 

먼저 데크를 사용하기위해서 import 해주고 데크의 길이와 뽑으려는 수의 횟수를 n,m의 변수로 받는다.

그리고 리스트의 형태로 뽑으려는 수를 순서대로 p의 변수로 받는다.

from collections import deque
import sys

n, m = map(int, sys.stdin.readline().split())
p = list(map(int, sys.stdin.readline().split())) #뽑으려는 수위치

 

다음은 데크(deque)를 활용해서 n길이 만큼의 데크를 dq의 변수로 받는다

dq = deque([i for i in range(1, n+1)])

 

이제 수를 뽑는 함수와 아닐 경우 왼쪽을 추가할지 오른쪽을 추가할지에 관해 for 문과 while문을 활용해서 구현해준다.

count = 0  #2,3의 연산이 얼마나 실행되는지 기록

for i in p: #p안에 있는 순서대로
    while True:
        if dq[0] == i: #dq[0]번째 수가 i면 1번 실행
            dq.popleft()
            break
        else:
            if dq.index(i) < len(dq)/2: #dq의절반보다 현재 i의 위치가 작다면 실행
                while dq[0] != i: #i와 같아질때 까지 2번 실행
                    dq.append(dq.popleft())
                    count += 1
            else:
                while dq[0] != i: #i와 같아질때 까지 3번 실행
                    dq.appendleft(dq.pop())
                    count +=1

print(count)

끝