Backjoon - problem 2738 행렬 덧셈 파이썬

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

2738번: 행렬 덧셈

 

 

• 입력 과 출력

 

 

• 해설

N,M = map(int , input().split())
nmList =[]
x,y = [],[]
for i in [x,y]:
    for j in range(N):
        i.append(list(map(int ,input().split())))

res = list(zip(x,y))
for i,j in enumerate(res):
    nmList.append([j[0][i]+j[1][i] for i in range(N)])

for matrix in nmList:
    print(' '.join(map(str,matrix)))

 

첫 번째 for문까지는 2차원 리스트를 입력 받아 만들어 주는 부분이다.

 

 

이 문제에서는 zip을 활용하여 

res = [([1, 1, 1], [3, 3, 3]), ([2, 2, 2], [4, 4, 4]), ([0, 1, 0], [5, 5, 100])]

이런식의 구조를 만들어 

각각의 튜플을 더하는 구조로 만들었다.