BOJ 1021 : 회전하는 큐 (Python)

문제

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

1021번: 회전하는 큐

 

 

코드

from collections import deque

n, m = map(int, input().split(' '))

array = list(map(int, input().split(' ')))

numlist = deque((i+1) for i in range(n))

result = 0
idx = 0

for i in range(m):
    for j in range(n):
        if array[i] == numlist[j]:
            idx = j
            break
    if idx == 0:
        numlist.popleft()
    elif idx == n-1:
        x = numlist.pop()
        numlist.appendleft(x)
        numlist.popleft()
        result += 1
    elif idx <= (n//2):
        for j in range(idx):
            result += 1
            temp = numlist.popleft()
            numlist.append(temp)
        numlist.popleft()
    else:
        for j in range(n-idx):
            result += 1
            temp = numlist.pop()
            numlist.appendleft(temp)
        numlist.popleft()
    n -= 1

print(result)

 

 

풀이

 

덱을 이용해 풀었다.

추출해내고 싶은 값이 전체 원소 개수에서 중앙이나 중앙에서 왼쪽에 위치할 경우에는 추출해내고 싶은 값이 인덱스 0 자리에 올 때까지 deque 모듈의 popleft() 를 실행한다. 실행 횟수는 idx 번이다. (실행할 때마다 횟수를 체크해야하므로 result 값에 1을 더해준다.) 그리고 곧바로 appendleft() 함수로 그 값을 다시 배열에 앞쪽에 넣어줘야 한다.

추출해내고 싶은 값이 인덱스 0 자리에 오게 되면 popleft()로 추출해낸다. 이때는 2번, 3번 연산에 해당하지 않으므로 result 에 1을 더해주지 않는다!

 

 

추출해내고 싶은 값이 전체 원소 개수에서 중앙보다 오른쪽에 위치할 경우에는 추출해내고 싶은 값이 인덱스 0 자리에 올 때까지 deque 모듈의 pop() 를 실행한다. 실행 횟수는 n - idx 번이다.(실행할 때 마다 횟수를 체크해야하므로 result 값에 1을 더해준다.) 그리고 곧바로 appendleft() 함수로 그 값을 다시 배열에 앞쪽에 넣어줘야 한다.

이 또한 위와 같이 추출해내고 싶은 값이 인덱스 0 자리에 오게 되면 popleft()로 추출해낸다. 이때는 2번, 3번 연산에 해당하지 않으므로 result 에 1을 더해주지 않는다!

 

 

헷갈렸던 점

추출해내고 싶은 값이 원소의 마지막 자리에 있으면 거기서 바로 pop() 하면 된다고 생각했는데 문제를 잘 못 읽었던 것이었다.