[백준][Python] 17123 배열놀이

처음에 그냥 matrix에 넣었다가 시간초과나고

합연산은 한번에 하는걸로 코드 수정. 통과

import sys
input = sys.stdin.readline



for test in range(int(input())):
    N,M = map(int,input().split())
    matrix = [list(map(int,input().split())) for _ in range(N)]
    ans = [[] for _ in range(2)]
    for i in range(N):
        ans[0].append(sum(matrix[i]))
    for i in range(N):
        ans[1].append(sum(matrix[m][i] for m in range(N)))
    for _ in range(M):
        r1,c1,r2,c2,v = map(int,input().split())
        # c1~c2행까지 (r2-r1)*v만큼 증가.
        for i in range(r1-1,r2):
            ans[0][i] += (c2-c1+1)*v
        # r1~r2열까지 (c2-c1)*v만큼 증가.
        for i in range(c1-1,c2):
            ans[1][i] += (r2-r1+1)*v

    for i in range(2):
        print(*ans[i])