[백준/BOJ] 2738번: 행렬 덧셈 (Python 파이썬)

문제

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

 

입출력 규칙

1. 입력

첫째 줄에 행렬의 크기 N 과 M이 주어진다. 둘째 줄부터 N개의 줄에 행렬 A의 원소 M개가 차례대로 주어진다. 이어서 N개의 줄에 행렬 B의 원소 M개가 차례대로 주어진다. N과 M은 100보다 작거나 같고, 행렬의 원소는 절댓값이 100보다 작거나 같은 정수이다.

2. 출력
첫째 줄부터 N개의 줄에 행렬 A와 B를 더한 행렬을 출력한다. 행렬의 각 원소는 공백으로 구분한다.

 

---

문제 풀이

<내 풀이1>

 

2개의 2차원 배열(행렬)을 입력받아 각 요소를 더한 결과를 출력하는 코드이다.

아마, 가장 기본 코드이지 않을까 싶다.

n, m = map(int,input().split())

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

for i in range(n):
    for j in range(m):
        print(A[i][j]+B[i][j],end=' ')
    print()

 

<숏코딩>

 

조금 더 간결한 코드이다. 

특히, 결과를 출력하는 코드가 아주 심플하다.

 

• a라는 리스트를 초기화한다. 해당 리스트는 두 행렬의 모든 요소를 저장한다.
• zip(a[:n*m], a[n*m:])을 이용하여, 두 행렬의 동일 위치에 있는 요소들을 쌍으로 묶는다.
• 그리고, map(sum,zip(...) ) 을 이용하여, 각 쌍의 요소를 더한다.

n,m=map(int,input().split())
a = []
for i in range(n*2):
    a += list(map(int,input().split()))

print(*map(sum,zip(a[:n*m],a[n*m:])))