[백준]28438 행렬연산(행렬 계산하기) python

from sys import stdin


input = stdin.readline

N,M,Q = map(int,input().split(" "))


row = {i:0 for i in range(N+1)}
col = {i:0 for i in range(M+1)}

for i in range(Q):
    n1,n2,n3 = map(int,input().split(" "))
    if n1 == 1:
        row[n2] += n3
    elif n1 == 2:
        col[n2] += n3

for i in range(1,N+1):
    for j in range(1,M+1):
        print(row[i]+col[j],end=" ")
    print()

풀이방법

n1 은 행 더하기인지 n2는 인덱스 n3는 더해지는 값으로

 

마지막 출력할 때 i행 j열의 더해진 n3를 계산하기만 하면 됨