[백준 2738번. 파이썬] 행렬 덧셈

📌 문제

 

📌 풀이

 

map() 함수를 꼭 외워놔야한다. 유용하니..(애초에 숫자는 int로 들어가게 해주세요오)

 

행과 열의 개수를 받고 빈 리스트 2개 생성.

 

행의 수만큼 반복하면서 각각 행렬 2개를 받은 후,

각 행의 각 열 단계로 반복하면서 요소들을 더하면 끝

 

A, B = map(int, input().split())
matrix1, matrix2 = [],[]

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

for j in range(A):
    row = list(map(int, input().split()))
    matrix2.append(row)

for row in range(A):
    for col in range(B):
        print(matrix1[row][col] + matrix2[row][col], end=' ')
    print()