[PYTHON/파이썬] 2738 행렬 덧셈

 

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

2738번: 행렬 덧셈

solution - 성공

import sys
N,M = map(int, sys.stdin.readline().strip().split())
one_matrix = [list(map(int, sys.stdin.readline().strip().split())) for _ in range(N)] # [[1, 1, 1], [2, 2, 2], [0, 1, 0]]
two_matrix = [list(map(int, sys.stdin.readline().strip().split())) for _ in range(N)] # [[1, 1, 1], [2, 2, 2], [0, 1, 0]]

# 0으로 이뤄진 이차원 리스트 생성
result = [[0]*M for _ in range(N)]
# [[0, 0, 0], [0, 0, 0], [0, 0, 0]]

for i in range(N):
  for j in range(M):
    # 각 요소를 더해 result 요소에 값을 갱신함.
    result[i][j] = one_matrix[i][j] + two_matrix[i][j]
  # 각 행 출력함.
  print(*result[i])

---

만약, 행렬의 곱셈도 궁금하시다면 아래의 글을 참고해주세요

2023.02.09 - [algorithm/programmers] - [Level2] 행렬의 곱셈 - 3중 for문 / zip

[Level2] 행렬의 곱셈 - 3중 for문 / zip