[Python] 백준 2738번 : 행렬 덧셈

코드

a, b = [], []

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

for i in range(N):
    row = list(map(int, input().split()))
    a.append(row)

for i in range(N):
    row = list(map(int, input().split()))
    b.append(row)

for x in range(N):
    for y in range(M):
        print(a[x][y] + b[x][y], end=' ')
    print()

설명

a, b에 행렬을 받을 리스트를 만들어준다.

N, M에 행렬의 크기를 입력받는다.

반복문을 통해 N만큼 반복하여 list형태로a,b에 행렬의 원소들을 입력받는다.

 

이중반복문을 통해 모든 행렬을 돈다.

xy 좌표를 통해 행렬을 하나씩 돌면서 각 1대1 매칭으로 a와 b에 있는 값들을 더해준다.