BOJ1021 회전하는 큐

randi65535·2020년 12월 6일
0

0, 1~N-1일때 잘 구별해 줘야 함

i랑 idx랑 헷갈렸다. 특별한 이유가 없는 이상은 같이 쓰지말자.

from collections import deque
import sys
input = sys.stdin.readline

N, M = map(int, input().strip().split())

ms = list(map(int, input().strip().split()))
elems = [i+1 for i in range(N)]
dq = deque(elems)

ans = 0
for m in ms:
	i = dq.index(m)

	if i == 0:
		dq.popleft()
	else:
		if i < len(dq)-i:
			dq.rotate(-i)
			ans += i

		else:
			dq.rotate(len(dq)-i)
			ans += len(dq)-i
		dq.popleft()
print(ans)
profile
unsinged int 8byte-1

0개의 댓글

관련 채용 정보