1021 회전하는 큐

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

1021번: 회전하는 큐

 

내 풀이(맞음)

from collections import deque

n, m = map(int, input().split())
ary = list(map(int, input().split()))
num = deque()
for i in range(n):
    num.append(i+1)
l = 0
id = 0
count = 0
while(True):
    if ary[l] == num[0]:
        num.popleft()
        l += 1
        if l == m:
            break
    else:
        t = num.index(ary[l])
        if t <= len(num)//2:
            temp = num.popleft()
            num.append(temp)
            count += 1
        else:
            temp = num.pop()
            num.appendleft(temp)
            count += 1
print(count)

덱을 사용했다. l은 ary에서 몇번째 값을 뽑아낼 것인지 확인하는 것이다. count는 움직인 회수이다. while문에서 만약 ary[l]과 num[0]이 같으면 num의 왼쪽 수를 삭제하고 l에 1을 더한다. 만약 l과 m이 같으면 멈춘다. 같지 않다면 t에 찾아야 할 ary[l]번째의 인덱스 값을 저장한다. 그리고 t가 len(num)//2보다 작거나 같으면 num에서 숫자를 왼쪽에서 오른쪽으로 보낸다. len(num)//2보다 크면 오른쪽에서 왼쪽으로 숫자를 보내고 count에 1을 더한다.