[백준] 2738. 행렬 덧셈

N*M크기의 두 행렬 A와 B가 주어졌을 때, 두 행렬을 더하는 프로그램을 작성하라.

선형대수를 공부하면서 만들었던 아래 함수를 이용하면 여러 행렬의 합을 쉽게 구할 수 있다.

def add(*a):
    return [[sum(v) for v in zip(*i)] for i in zip(*a)]


m, n = (int(x) for x in input().split())
a, b = ([[int(n) for n in input().split()] for _ in range(m)] for _ in range(2))

for r in add(a, b):
    for v in r:
        print(v)

[백준] 2738. 행렬 덧셈

Reference¶