[백준] 2838 행렬 덧셈

문제

2738번: 행렬 덧셈

풀이

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

list_a = []
list_b = []

for _ in range(n):
    temp_list = list(map(int, input().split()))
    list_a = list_a + temp_list

for _ in range(n):  
	#처음에 m으로 작성하는 바람에 정방행렬이 아닌 경우 오류 발생
    temp_list = list(map(int, input().split()))
    list_b = list_b + temp_list
    
k = 1
for a, b in zip(list_a, list_b):
    if k < m:
      print(a+b, end=' ')
    else:
      print(a+b)
      k=0
    k=k+1

Point

• N*M 행렬이라고 했을 때, 예제 입력 값만 보고 정방행렬이라고 판단한 부분
  > 행렬의 열 입력받는 횟수를 m으로 두어 런타임 오류 발생
• 두 리스트의 각 원소를 더하는데 사용될 수 있는 함수(zip)을 사용하여 코드 단축