백준 2738 행렬 덧셈/파이썬

📌문제

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

 

📌풀이

a,b라는 일차원 배열을 미리 생성

한줄씩 입력 받을 때마다 새로운 배열을 생성한 후 기존 일차원 배열에 append하여 2차원 배열 생성

두번 반복하여 2차원 배열 a,b가 만들어지면 두 배열의 합을 구함

 

📌코드

n, m = map(int, input().split())
a, b = [], []

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 i in range(n):
    for j in range(m):
        result = a[i][j] + b[i][j]
        print(result, end=" ")
    print()