2738:행렬 덧셈(python)

https://www.acmicpc.net/problem/2738

2738번: 행렬 덧셈

문제

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

풀이

행렬의 크기를 정해주고 append 시킨다음 인덱스 번호를 잘 지정해서 반복문으로 더해주면 된다.

코드

 

# MxN 행렬의 크기 입력 받기
M, N = map(int, input().split())
A, B = [], []

# 행렬 A 입력 받기
for _ in range(M):
    row = list(map(int, input().split()))
    A.append(row)

# 행렬 B 입력 받기
for _ in range(M):
    row = list(map(int, input().split()))
    B.append(row)

# 두 행렬의 합 구하기
for row in range(M):
    for col in range(N):
        print(A[row][col] + B[row][col], end=' ')
    print()  # 열 바꿈