BOJ 2738번: 행렬 덧셈

핵심 포인트

• 2차원 배열
• 리스트 컴프리헨션

제출코드

N, M = map(int, input().split())
m1 = [list(map(int, input().split())) for _ in range(N)]
m2 = [list(map(int, input().split())) for _ in range(N)]
m3 = [[0 for _ in range(M)] for _ in range(N)]

for i in range(N):
  for j in range(M):
    m3[i][j] = m1[i][j] + m2[i][j]

for row in m3:
  print(' '.join(map(str, row)))

해설

첫번째 줄에는 두 행렬의 크기 N과 M을 입력받습니다. 다음 N개의 줄에서는 행렬 m1의 값들을 리스트 컴프리헨션을 이용해서 입력받고 다음은 m2도 입력받습니다. m3는 두 행렬 m1과 m2를 더하고 값을 저장하는 2차원 행렬로 m1과 m2와 동일한 크기를 가지게 됩니다. 

 

다음에는 이중 반복문을 이용해서 각 성분의 합을 m3에 저장해준뒤 각 행별로 출력해주면 됩니다. 

참고자료 및 그림출처

백준 코딩 문제