[백준/Python] 2738번 행렬 덧셈 문제

■ 2738번 행렬 덧셈 문제

 

출처: 백준 2738번 행렬 덧셈 문제

 

■ 코드 풀이

 

크게 어렵지 않은 문제입니다. 우선 행렬 A와 B의 row 개수는 N으로 동일하므로 입력을 2*N번만큼 한 번에 받아서 temp라는 list에 append 합니다. 그리고 다시 for loop를 돌면서 list의 element-wise 덧셈을 실행하는 함수를 실행(map) 합니다. 출력은 end 옵션을 활용하고 한 줄의 연산이 끝나면 print()로 줄 바꿈 효과를 줍니다.

import sys

def Add_elementwise(elements):
	x, y = elements
	return x+y

N,M = map(int, sys.stdin.readline().split())
temp = []

for i in range(2*N):
	temp.append(list(map(int,  sys.stdin.readline().split())))

for i in range(N):
	for value in list(map(Add_elementwise, zip(temp[i],temp[i+N]))):
		print(value, end = " ")
	print()