[Goorm level] 회전 배열 (python)

회전 배열 - 구름LEVEL

구름LEVEL

 

 

배열을 돌리는 부분은 파이썬의 deque.rotate()를 사용하면 된다.

 

그리고 회전할 때의 조건을 잘 읽지 않아 삽질을 오래 했는데 (문제를 잘 읽어야 한다.)

 

매 회전할 때마다 리스트의 첫 값을 가져와 그 첫 값을 회전하는 것이다.

 

그걸 m번 반복하는 것이므로, 무조건 횟수가 m으로 지정되어 있는 것이 아니다.

 

공교롭게도 배열을 회전하는 횟수를 m으로만 지정해도 테스트케이스가 통과되어서 맞왜틀을 계속 외치고 있었는데, 문제를 여러 번 읽어보니 내가 잘못 구현하고 있었음을 깨달았다.

 

from collections import deque

n, m = map(int, input().split())
numbers = list(map(int, input().split()))
q = deque(numbers)

for _ in range(m):
    d = q[0]
    q.rotate(-d)

print(q[0])

 

references

회전 배열 - 구름LEVEL

구름LEVEL