인프런 커뮤니티 질문&답변

풀이를 보기전에 파이썬 내장라이브러리인 collections.deque를 이용하여  아래와 같은 함수를 만들어서 풀었습니다.

괜찮은 풀이인지 확인받고싶습니다.

그리고 실제 코딩테스트에서 collections, itertools같은 내장라이브러리를 사용해도되나요?

from collections import deque


def move_list(direct, many, l):
    sub_q = deque()
    q = deque(l)

    if direct == left:
        for _ in range(many):
            sub_q.append(q.popleft())
        return list(q+sub_q)

    if direct == right:
        for _ in range(n-many, n):
            sub_q.appendleft(q.pop())
        return list(sub_q+q)


left, right = 0, 1
n = int(input())

matrix = [list(map(int, input().split())) for _ in range(n)]
m = int(input())

for _ in range(m):
    row, direct, many = map(int, input().split())
    many = many % n

    for i in range(n):
        if i == row-1:
            matrix[i] = move_list(direct, many, matrix[i])

s, e = 0, n-1
res = 0

for i in range(n):
    for j in range(s, e+1):
        res += matrix[i][j]

    if i < n // 2:
        s += 1
        e -= 1
    else:
        s -= 1
        e += 1

print(res)