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

채점기 돌려가면서 코드를 수정하여 이런 코드를 작성하였습니다.

혹시 이 코드가 잘못되거나 반례가 있을까요??

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

for _ in range(m):
    s, b, c = map(int, input().split())
    tmp = [0] * n
    for i in range(n):
        if b == 0:
            if c > n:
                tmp[n - c + i] = a[s - 1][i]
            else:
                tmp[i - c] = a[s - 1][i]
        if b == 1:
            t = i + c
            if t < n:
                tmp[t] = a[s - 1][i]
            else:
                tmp[t - n] = a[s - 1][i]
    a[s - 1] = tmp

s = 0
e = n
result = 0
for i in range(n):
    for j in range(s, e):
        result += a[i][j]
    if i < n // 2:
        s += 1
        e -= 1
    else:
        s -= 1
        e += 1

print(result)