[Python] 백준 2738 행렬 덧셈

 

문제분석

• 행렬 N*M 크기의 두 행렬 A, B가 주어졌을 때 두 행렬을 더하여 출력

 

코드 설계

• 각 행렬을 2차원 리스트로 받고
• 인덱스 같은 값끼리 더해서 출력

 

정답 코드

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
A = []
for _ in range(n) :
    A.append(list(map(int, input().split())))

B = []
for _ in range(n) :
    B.append(list(map(int, input().split())))

ans = [[0] * m for _ in range(n)]
for r in range(n) :
    for c in range(m) :
        ans[r][c] = A[r][c] + B[r][c]

for r in range(n) :
    for c in range(m) :
        print(ans[r][c], end=' ')
    print()

 

다른 사람 코드 1

A, B = [], []

N, M = map(int, input().split())

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

for row in range(N):
    row = list(map(int, input().split()))
    B.append(row)
    
for row in range(N):
    for col in range(M):
        print(A[row][col] + B[row][col], end=' ')
    print()

그냥 나는 ans 리스트를 만들어서 출력했는데 이 코드는 그냥 바로 덧셈해서 출력