본문 바로가기

파이썬 알고리즘 (인프런)/탐색&시물레이션

8. 곳감

# 곳감 (모래시계)

def rotate(N, MP, r, d, c):
    a = MP[r-1][:]
    for i in range(N):
        if d == 0:
            MP[r-1][i] = a[(i+c) % N]
        else:
            MP[r-1][i] = a[(i-c) % N]
    return MP


def summ(N, MP):
    tot = 0
    s, e = 0, N
    for i in range(N):
        tot += sum(MP[i][s:e])
        if i < N // 2:
            s += 1
            e -= 1
        else:
            s -= 1
            e += 1
    return tot


N = int(input())
MP = [list(map(int, input().split())) for _ in range(N)]
M = int(input())
for _ in range(M):
    r, d, c = map(int, input().split())
    MP = rotate(N, MP, r, d, c)
print(summ(N, MP))

'파이썬 알고리즘 (인프런) > 탐색&amp;amp;시물레이션' 카테고리의 다른 글

10. 스도쿠 검사  (0) 2021.02.09
9. 봉우리  (0) 2021.02.09
7. 사과나무  (0) 2021.02.09
6. 격자판 최대합  (0) 2021.02.09
5. 수들의 합  (0) 2021.02.09